Update src/diffusers/loaders/single_file_model.py

Co-authored-by: Dhruv Nair <[email protected]>
huggingface · Dec 23, 2024 · 6720c51 · 6720c51
1 parent a6e3745
commit 6720c51
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/diffusers/loaders/single_file_model.py b/src/diffusers/loaders/single_file_model.py
@@ -362,7 +362,7 @@ def from_single_file(cls, pretrained_model_link_or_path_or_dict: Optional[str] =
 
         else:
             _, unexpected_keys = model.load_state_dict(
-                diffusers_format_checkpoint, strict=False, disable_mmap=disable_mmap
+                diffusers_format_checkpoint, strict=False
             )
 
         if model._keys_to_ignore_on_load_unexpected is not None: