Do not download onnx model in sd pipeline if from_onnx=False (#402)

eaidova · web-flow · commit 54d5d7967704 · 2023-08-09T17:43:08.000+09:00
diff --git a/optimum/intel/openvino/modeling_diffusion.py b/optimum/intel/openvino/modeling_diffusion.py
@@ -217,6 +217,9 @@ def _from_pretrained(
                     cls.config_name,
                 }
             )
+            ignore_patterns = ["*.msgpack", "*.safetensors", "*pytorch_model.bin"]
+            if not from_onnx:
+                ignore_patterns.extend(["*.onnx", "*.onnx_data"])
             # Downloads all repo's files matching the allowed patterns
             model_id = snapshot_download(
                 model_id,
@@ -225,7 +228,7 @@ def _from_pretrained(
                 use_auth_token=use_auth_token,
                 revision=revision,
                 allow_patterns=allow_patterns,
-                ignore_patterns=["*.msgpack", "*.safetensors", "*pytorch_model.bin"],
+                ignore_patterns=ignore_patterns,
             )
         new_model_save_dir = Path(model_id)
 

Original file line number	Diff line number	Diff line change
`@@ -217,6 +217,9 @@ def _from_pretrained(`
`217`	`217`	`cls.config_name,`
`218`	`218`	`}`
`219`	`219`	`)`
	`220`	`+ ignore_patterns = [".msgpack", ".safetensors", "*pytorch_model.bin"]`
	`221`	`+ if not from_onnx:`
	`222`	`+ ignore_patterns.extend([".onnx", ".onnx_data"])`
`220`	`223`	`# Downloads all repo's files matching the allowed patterns`
`221`	`224`	`model_id = snapshot_download(`
`222`	`225`	`model_id,`
`@@ -225,7 +228,7 @@ def _from_pretrained(`
`225`	`228`	`use_auth_token=use_auth_token,`
`226`	`229`	`revision=revision,`
`227`	`230`	`allow_patterns=allow_patterns,`
`228`		`- ignore_patterns=[".msgpack", ".safetensors", "*pytorch_model.bin"],`
	`231`	`+ ignore_patterns=ignore_patterns,`
`229`	`232`	`)`
`230`	`233`	`new_model_save_dir = Path(model_id)`
`231`	`234`