From 8c296d36f1f9c795bf2a330c6e62381751ca0ed1 Mon Sep 17 00:00:00 2001 From: Bas Krahmer Date: Thu, 5 Oct 2023 10:28:23 +0200 Subject: [PATCH] Ignore .xml files for Stable Diffusion ORT downloads (#1428) Co-authored-by: bas --- optimum/onnxruntime/modeling_diffusion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/optimum/onnxruntime/modeling_diffusion.py b/optimum/onnxruntime/modeling_diffusion.py index 123bcdbd926..bc0d17a93aa 100644 --- a/optimum/onnxruntime/modeling_diffusion.py +++ b/optimum/onnxruntime/modeling_diffusion.py @@ -315,7 +315,7 @@ def _from_pretrained( use_auth_token=use_auth_token, revision=revision, allow_patterns=allow_patterns, - ignore_patterns=["*.msgpack", "*.safetensors", "*.bin"], + ignore_patterns=["*.msgpack", "*.safetensors", "*.bin", "*.xml"], ) new_model_save_dir = Path(model_id)