Skip to content

Commit 0e50401

Browse files
authored
[Single file] Support revision argument when loading single file config (#10168)
update
1 parent 6131a93 commit 0e50401

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/diffusers/loaders/single_file_model.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -219,7 +219,7 @@ def from_single_file(cls, pretrained_model_link_or_path_or_dict: Optional[str] =
219219
mapping_functions = SINGLE_FILE_LOADABLE_CLASSES[mapping_class_name]
220220

221221
checkpoint_mapping_fn = mapping_functions["checkpoint_mapping_fn"]
222-
if original_config:
222+
if original_config is not None:
223223
if "config_mapping_fn" in mapping_functions:
224224
config_mapping_fn = mapping_functions["config_mapping_fn"]
225225
else:
@@ -243,7 +243,7 @@ def from_single_file(cls, pretrained_model_link_or_path_or_dict: Optional[str] =
243243
original_config=original_config, checkpoint=checkpoint, **config_mapping_kwargs
244244
)
245245
else:
246-
if config:
246+
if config is not None:
247247
if isinstance(config, str):
248248
default_pretrained_model_config_name = config
249249
else:
@@ -270,6 +270,7 @@ def from_single_file(cls, pretrained_model_link_or_path_or_dict: Optional[str] =
270270
subfolder=subfolder,
271271
local_files_only=local_files_only,
272272
token=token,
273+
revision=revision,
273274
)
274275
expected_kwargs, optional_kwargs = cls._get_signature_keys(cls)
275276

0 commit comments

Comments
 (0)