Avoid duplicating PyTorch + safetensors downloads. (#1836)

pcuenca · web-flow · commit 68e24259af62 · 2022-12-27T14:58:15.000+01:00
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(

Original file line number	Diff line number	Diff line change
`@@ -495,6 +495,8 @@ def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union[str, os.P`
`495`	`495`	`)`
`496`	`496`	`if is_safetensors_compatible(info):`
`497`	`497`	`ignore_patterns.append("*.bin")`
	`498`	`+ else:`
	`499`	`+ ignore_patterns.append("*.safetensors")`
`498`	`500`
`499`	`501`	`# download all allow_patterns`
`500`	`502`	`cached_folder = snapshot_download(`