Skip to content

Commit bcbd493

Browse files
committed
format
1 parent d1c4a61 commit bcbd493

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

src/diffusers/models/modeling_utils.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1167,9 +1167,7 @@ def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union[str, os.P
11671167
state_dict = None
11681168
if not is_sharded:
11691169
# Time to load the checkpoint
1170-
state_dict = load_state_dict(
1171-
resolved_model_file[0], disable_mmap=disable_mmap, dduf_entries=dduf_entries
1172-
)
1170+
state_dict = load_state_dict(resolved_model_file[0], disable_mmap=disable_mmap, dduf_entries=dduf_entries)
11731171
# We only fix it for non sharded checkpoints as we don't need it yet for sharded one.
11741172
model._fix_state_dict_keys_on_load(state_dict)
11751173

0 commit comments

Comments
 (0)