Skip to content

Commit 68dd3ff

Browse files
committed
fix save pretrained
1 parent 3c39f4d commit 68dd3ff

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

optimum/intel/openvino/modeling_visual_language.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -324,7 +324,7 @@ def _save_pretrained(self, save_directory: Union[str, Path]):
324324
dst_file_names = [
325325
"openvino_language_model.xml",
326326
"openvino_text_embeddings_model.xml",
327-
"openvino_vision_embeddings.xml",
327+
"openvino_vision_embeddings_model.xml",
328328
]
329329
for part in self.additional_parts:
330330
model = getattr(self, f"{part}_model", None)

0 commit comments

Comments
 (0)