Avoid duplicating PyTorch + safetensors downloads. (#1836)

68e24259 · Pedro Cuenca · GitHub · 1f1b6c65 · 68e24259
Unverified Commit 68e24259 authored Dec 27, 2022 by Pedro Cuenca Committed by GitHub Dec 27, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

src/diffusers/pipeline_utils.py src/diffusers/pipeline_utils.py +2 -0

No files found.
--- a/src/diffusers/pipeline_utils.py
+++ b/src/diffusers/pipeline_utils.py
@@ -495,6 +495,8 @@ class DiffusionPipeline(ConfigMixin):
                )
                if is_safetensors_compatible(info):
                    ignore_patterns.append("*.bin")
+                else:
+                    ignore_patterns.append("*.safetensors")

            # download all allow_patterns
            cached_folder = snapshot_download(