From 68e24259af627d2e32b5f06e41c095a511b16aca Mon Sep 17 00:00:00 2001 From: Pedro Cuenca Date: Tue, 27 Dec 2022 14:58:15 +0100 Subject: [PATCH] Avoid duplicating PyTorch + safetensors downloads. (#1836) --- src/diffusers/pipeline_utils.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/diffusers/pipeline_utils.py b/src/diffusers/pipeline_utils.py index 23c4b29a53..50fbbedc99 100644 --- 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(