Open purejomo opened 2 months ago
Hi @purejomo,
I was unfortunately not able to reproduce the error you're describing. Yes _SENTENCE_TRANSFORMERS_TASKS_TO_MODEL_LOADERS
is defined in https://github.com/huggingface/optimum/blob/v1.21.4/optimum/exporters/tasks.py#L201-L204, could it be that in your case is_torch_available
https://github.com/huggingface/optimum/blob/v1.21.4/optimum/exporters/tasks.py#L156 returns False?
System Info
In "task.py" file, I touhgt _SENTENCE_TRANSFORMERS_TASKS_TO_MODEL_LOADERS is defined already.
Who can help?
@philschmid @michaelbenayoun @JingyaHuang, @echarlaix
Information
Tasks
examples
folder (such as GLUE/SQuAD, ...)Reproduction (minimal, reproducible, runnable)
python 3.10.14 torch 2.4.0+cu121 optimum 1.21.4
onnx 1.16.2
onnxruntime 1.19.0
transformers 4.43.4
Expected behavior
error should be figured out. I'm trying to convert hf-llama model to onnx for graph optimization