Model mistralai/Mistral-7B-Instruct-v0.3 doesn't have defined tasks #1409
Description
While exporting model mistralai/Mistral-7B-Instruct-v0.3 there is not default task defined:
optimum-cli export openvino -m mistralai/Mistral-7B-Instruct-v0.3 output
Traceback (most recent call last):
File "/home/dtrawins/model_server/demos/common/export_models/.venv/lib/python3.12/site-packages/optimum/exporters/openvino/__main__.py", line 83, in infer_task
task = TasksManager._infer_task_from_model_name_or_path(
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/dtrawins/model_server/demos/common/export_models/.venv/lib/python3.12/site-packages/optimum/exporters/tasks.py", line 1780, in _infer_task_from_model_name_or_path
raise KeyError(f"Could not find the proper task name for the model {model_name_or_path}.")
KeyError: 'Could not find the proper task name for the model mistralai/Mistral-7B-Instruct-v0.3.'
Could it be defined like for other public models list meta-llama/Llama-3.2-3B-Instruct ?
Metadata
Metadata
Assignees
Labels
No labels