From a3bb3446db27b66c83ad964518a59b73241e50c3 Mon Sep 17 00:00:00 2001 From: Lucain Date: Tue, 4 Jun 2024 14:26:02 +0200 Subject: [PATCH] Deprecate resume_download (#586) --- optimum/neuron/distributed/utils.py | 2 +- optimum/neuron/utils/misc.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/optimum/neuron/distributed/utils.py b/optimum/neuron/distributed/utils.py index 7dd4aaaf1..66529369b 100644 --- a/optimum/neuron/distributed/utils.py +++ b/optimum/neuron/distributed/utils.py @@ -1124,7 +1124,7 @@ def from_pretrained_for_mp( kwargs.pop("state_dict", None) kwargs.pop("from_tf", False) kwargs.pop("from_flax", False) - resume_download = kwargs.pop("resume_download", False) + resume_download = kwargs.pop("resume_download", None) proxies = kwargs.pop("proxies", None) kwargs.pop("output_loading_info", False) kwargs.pop("use_auth_token", None) diff --git a/optimum/neuron/utils/misc.py b/optimum/neuron/utils/misc.py index 28111afac..5f399380c 100644 --- a/optimum/neuron/utils/misc.py +++ b/optimum/neuron/utils/misc.py @@ -239,7 +239,7 @@ def download_checkpoints_in_cache( kwargs.pop("state_dict", None) from_tf = kwargs.pop("from_tf", False) from_flax = kwargs.pop("from_flax", False) - resume_download = kwargs.pop("resume_download", False) + resume_download = kwargs.pop("resume_download", None) proxies = kwargs.pop("proxies", None) kwargs.pop("output_loading_info", False) kwargs.pop("use_auth_token", None)