Fix for use_safetensors parameters, allow use of parameter on loading…

… submodels (#9576)
huggingface · Oct 4, 2024 · b185955 · b185955
1 parent 751893d
commit b185955
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/image.png b/image.png
diff --git a/src/diffusers/models/modeling_utils.py b/src/diffusers/models/modeling_utils.py
@@ -526,7 +526,7 @@ def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union[str, os.P
         use_safetensors = kwargs.pop("use_safetensors", None)
 
         allow_pickle = False
-        if use_safetensors is None or use_safetensors:
+        if use_safetensors is None:
             use_safetensors = True
             allow_pickle = True