The checkpoint is not correctly tagged with the library name
#1
by
Jingya
HF staff
- opened
The checkpoint i tagged with generic whereas it should be transformers, this lead to export failure using the Optimum library:
File "/home/ubuntu/py_venv/optimum_ort_1.13/lib/python3.8/site-packages/optimum/exporters/tasks.py", line 1148, in get_model_class_for_task
tasks_to_model_loader = TasksManager._LIBRARY_TO_TASKS_TO_MODEL_LOADER_MAP[library]
KeyError: 'generic'