diff --git a/src/diffusers/loaders/single_file_model.py b/src/diffusers/loaders/single_file_model.py index 08a47143f7..a2f27b765a 100644 --- a/src/diffusers/loaders/single_file_model.py +++ b/src/diffusers/loaders/single_file_model.py @@ -21,8 +21,8 @@ import torch from huggingface_hub.utils import validate_hf_hub_args from typing_extensions import Self -from ..quantizers import DiffusersAutoQuantizer from .. import __version__ +from ..quantizers import DiffusersAutoQuantizer from ..utils import deprecate, is_accelerate_available, logging from .single_file_utils import ( SingleFileComponentError, diff --git a/src/diffusers/loaders/single_file_utils.py b/src/diffusers/loaders/single_file_utils.py index f1f8dbe125..b55b1b5520 100644 --- a/src/diffusers/loaders/single_file_utils.py +++ b/src/diffusers/loaders/single_file_utils.py @@ -405,11 +405,11 @@ def load_single_file_checkpoint( local_files_only=None, revision=None, disable_mmap=False, - user_agent=None + user_agent=None, ): if user_agent is None: user_agent = {"file_type": "single_file", "framework": "pytorch"} - + if os.path.isfile(pretrained_model_link_or_path): pretrained_model_link_or_path = pretrained_model_link_or_path