dunzhang

/

stella_en_400M_v5

like 111

Follow

StellaEncoder 16

Hi. I am trying to generate the ONNX version using the command: python -m optimum.exporters.onnx --model=/home/ubuntu/sentence_transformer/stella_en_400M_v5 --feature_size 1024 --task feature-extraction --trust-remote-code --device cuda .

But getting the error: RuntimeError: Found an unsupported argument type c10::SymInt in the JIT tracer. File a bug report.

Can someone help on this?