Deprecate resume_download (#586)

huggingface · Jun 4, 2024 · a3bb344 · a3bb344
1 parent a033e04
commit a3bb344
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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)