The checkpoint is not correctly tagged with the library name

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/", line 1148, in get_model_class_for_task
    tasks_to_model_loader = TasksManager._LIBRARY_TO_TASKS_TO_MODEL_LOADER_MAP[library]
KeyError: 'generic'

Sign up or log in to comment