Skip to content

GitLab

Unverified Commit 68e24259 authored by Pedro Cuenca's avatar Pedro Cuenca Committed by GitHub
Browse files

Avoid duplicating PyTorch + safetensors downloads. (#1836)

parent 1f1b6c65
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
src/diffusers/pipeline_utils.py
View file @ 68e24259
...@@ -495,6 +495,8 @@ class DiffusionPipeline(ConfigMixin): ...@@ -495,6 +495,8 @@ class DiffusionPipeline(ConfigMixin):
) )
if is_safetensors_compatible(info): if is_safetensors_compatible(info):
ignore_patterns.append("*.bin") ignore_patterns.append("*.bin")
else:
ignore_patterns.append("*.safetensors")
# download all allow_patterns # download all allow_patterns
cached_folder = snapshot_download( cached_folder = snapshot_download(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment