format

huggingface · Feb 14, 2025 · bcbd493 · bcbd493
1 parent d1c4a61
commit bcbd493
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/diffusers/models/modeling_utils.py b/src/diffusers/models/modeling_utils.py
@@ -1167,9 +1167,7 @@ def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union[str, os.P
         state_dict = None
         if not is_sharded:
             # Time to load the checkpoint
-            state_dict = load_state_dict(
-                resolved_model_file[0], disable_mmap=disable_mmap, dduf_entries=dduf_entries
-            )
+            state_dict = load_state_dict(resolved_model_file[0], disable_mmap=disable_mmap, dduf_entries=dduf_entries)
             # We only fix it for non sharded checkpoints as we don't need it yet for sharded one.
             model._fix_state_dict_keys_on_load(state_dict)