Conversion from Safetensors to ONNX failed

#29
by Islam-Mohamed - opened

I tried to convert "ibm-granite/granite-embedding-english-r2" but it throws out this error :

Traceback (most recent call last):
 File "/usr/local/lib/python3.10/runpy.py", line 196, in _run_module_as_main
   return _run_code(code, main_globals, None,
 File "/usr/local/lib/python3.10/runpy.py", line 86, in _run_code
   exec(code, run_globals)
 File "/home/user/app/transformers.js/scripts/convert.py", line 456, in <module>
   main()
 File "/home/user/app/transformers.js/scripts/convert.py", line 341, in main
   main_export(**export_kwargs)
 File "/usr/local/lib/python3.10/site-packages/optimum/exporters/onnx/main.py", line 305, in main_export
   model = TasksManager.get_model_from_task(
 File "/usr/local/lib/python3.10/site-packages/optimum/exporters/tasks.py", line 2258, in get_model_from_task
   model_class = TasksManager.get_model_class_for_task(
 File "/usr/local/lib/python3.10/site-packages/optimum/exporters/tasks.py", line 1553, in get_model_class_for_task
   loaded_library = importlib.import_module(library)
 File "/usr/local/lib/python3.10/importlib/__init__.py", line 126, in import_module
   return _bootstrap._gcd_import(name[level:], package, level)
 File "<frozen importlib._bootstrap>", line 1050, in _gcd_import
 File "<frozen importlib._bootstrap>", line 1027, in _find_and_load
 File "<frozen importlib._bootstrap>", line 1004, in _find_and_load_unlocked
ModuleNotFoundError: No module named 'sentence_transformers'

Sign up or log in to comment