Error when convert to ONXX with O4 optimization level

#5
by nobita3921 - opened

I'm using your export_onnx.py script to convert the model to O4 optimization level but It throw the exception:
ValueError: The library name could not be automatically inferred. If using the command-line, please provide the argument --library {transformers,diffusers,timm,sentence_transformers}. Example: --library diffusers.

This is my command:
python export_onnx.py --output onnx_model --opset 19 --device cuda --optimize O4 --atol 1e-5

Sign up or log in to comment