From edf5ba6a17d012411c1fe3ceaf24f71f1899bc48 Mon Sep 17 00:00:00 2001 From: Lucain Date: Fri, 24 May 2024 12:11:03 +0200 Subject: [PATCH] Respect `resume_download` deprecation V2 (#8267) * Fix resume_downoad FutureWarning * only resume download --- src/diffusers/loaders/single_file.py | 2 +- src/diffusers/loaders/single_file_model.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/diffusers/loaders/single_file.py b/src/diffusers/loaders/single_file.py index 041cb28e84..32b458ddd5 100644 --- a/src/diffusers/loaders/single_file.py +++ b/src/diffusers/loaders/single_file.py @@ -340,7 +340,7 @@ class FromSingleFileMixin: deprecate("original_config_file", "1.0.0", deprecation_message) original_config = original_config_file - resume_download = kwargs.pop("resume_download", False) + resume_download = kwargs.pop("resume_download", None) force_download = kwargs.pop("force_download", False) proxies = kwargs.pop("proxies", None) token = kwargs.pop("token", None) diff --git a/src/diffusers/loaders/single_file_model.py b/src/diffusers/loaders/single_file_model.py index bd63b513b1..f06f483274 100644 --- a/src/diffusers/loaders/single_file_model.py +++ b/src/diffusers/loaders/single_file_model.py @@ -166,7 +166,7 @@ class FromOriginalModelMixin: "`from_single_file` cannot accept both `config` and `original_config` arguments. Please provide only one of these arguments" ) - resume_download = kwargs.pop("resume_download", False) + resume_download = kwargs.pop("resume_download", None) force_download = kwargs.pop("force_download", False) proxies = kwargs.pop("proxies", None) token = kwargs.pop("token", None)