From 6569f76bc9912e1ce7c76a8c74fb65359349a661 Mon Sep 17 00:00:00 2001 From: bas Date: Wed, 4 Oct 2023 15:35:25 +0200 Subject: [PATCH] Ignore .xml files for Stable Diffusion ORT downloads --- 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)