From 63814b5882a9a69fa31f58c1e6c357056927de6e Mon Sep 17 00:00:00 2001 From: Aryan Date: Tue, 10 Dec 2024 08:24:01 +0100 Subject: [PATCH] update --- src/diffusers/loaders/single_file_model.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/diffusers/loaders/single_file_model.py b/src/diffusers/loaders/single_file_model.py index be3139057078..0f01dd942734 100644 --- a/src/diffusers/loaders/single_file_model.py +++ b/src/diffusers/loaders/single_file_model.py @@ -219,7 +219,7 @@ def from_single_file(cls, pretrained_model_link_or_path_or_dict: Optional[str] = mapping_functions = SINGLE_FILE_LOADABLE_CLASSES[mapping_class_name] checkpoint_mapping_fn = mapping_functions["checkpoint_mapping_fn"] - if original_config: + if original_config is not None: if "config_mapping_fn" in mapping_functions: config_mapping_fn = mapping_functions["config_mapping_fn"] else: @@ -243,7 +243,7 @@ def from_single_file(cls, pretrained_model_link_or_path_or_dict: Optional[str] = original_config=original_config, checkpoint=checkpoint, **config_mapping_kwargs ) else: - if config: + if config is not None: if isinstance(config, str): default_pretrained_model_config_name = config else: @@ -270,6 +270,7 @@ def from_single_file(cls, pretrained_model_link_or_path_or_dict: Optional[str] = subfolder=subfolder, local_files_only=local_files_only, token=token, + revision=revision, ) expected_kwargs, optional_kwargs = cls._get_signature_keys(cls)