Avoid duplicating PyTorch + safetensors downloads. (#1836)

huggingface · Dec 27, 2022 · 68e2425 · 68e2425
1 parent 1f1b6c6
commit 68e2425
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/diffusers/pipeline_utils.py b/src/diffusers/pipeline_utils.py
@@ -495,6 +495,8 @@ def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union[str, os.P
                 )
                 if is_safetensors_compatible(info):
                     ignore_patterns.append("*.bin")
+                else:
+                    ignore_patterns.append("*.safetensors")
 
             # download all allow_patterns
             cached_folder = snapshot_download(