fix save pretrained

eaidova · eaidova · commit 68dd3ff384a5 · 2024-10-15T13:43:46.000+04:00
diff --git a/optimum/intel/openvino/modeling_visual_language.py b/optimum/intel/openvino/modeling_visual_language.py
@@ -324,7 +324,7 @@ def _save_pretrained(self, save_directory: Union[str, Path]):
         dst_file_names = [
             "openvino_language_model.xml",
             "openvino_text_embeddings_model.xml",
-            "openvino_vision_embeddings.xml",
+            "openvino_vision_embeddings_model.xml",
         ]
         for part in self.additional_parts:
             model = getattr(self, f"{part}_model", None)

Original file line number	Diff line number	Diff line change
`@@ -324,7 +324,7 @@ def _save_pretrained(self, save_directory: Union[str, Path]):`
`324`	`324`	`dst_file_names = [`
`325`	`325`	`"openvino_language_model.xml",`
`326`	`326`	`"openvino_text_embeddings_model.xml",`
`327`		`- "openvino_vision_embeddings.xml",`
	`327`	`+ "openvino_vision_embeddings_model.xml",`
`328`	`328`	`]`
`329`	`329`	`for part in self.additional_parts:`
`330`	`330`	`model = getattr(self, f"{part}_model", None)`