Update src/diffusers/loaders/single_file_model.py

danhipke · DN6 · web-flow · commit 6720c51ae78f · 2024-12-22T22:59:26.000-08:00
Co-authored-by: Dhruv Nair &lt;dhruv.nair@gmail.com&gt;
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:

Original file line number	Diff line number	Diff line change
`@@ -362,7 +362,7 @@ def from_single_file(cls, pretrained_model_link_or_path_or_dict: Optional[str] =`
`362`	`362`
`363`	`363`	`else:`
`364`	`364`	`_, unexpected_keys = model.load_state_dict(`
`365`		`- diffusers_format_checkpoint, strict=False, disable_mmap=disable_mmap`
	`365`	`+ diffusers_format_checkpoint, strict=False`
`366`	`366`	`)`
`367`	`367`
`368`	`368`	`if model._keys_to_ignore_on_load_unexpected is not None:`