Set pipeline_tag
Set the task type to text2text-generation
, so that it can be automatically inferred by conversion scripts, such as the official script for onnx conversion in transformers.js.
Hello @agentlans ,
Thank you for the great model!
I was experimenting with the transformers.js library, which requires an ONNX model, and tried to convert this one using the official script that depends on optimum. However, the script was failing to automatically infer the task type from the model card, and specifying it manually via the --task
parameter resulted in an incomplete conversion (the decoder_model_merged.onnx file was not being generated).
Cloning this repository and setting the pipeline_tag
fixes this issue.
Resulting ONNX models here.
Also, let me know if you have interest in cloning the ONNX repo, so it resides under your account (and I can delete mine after) :)
Thanks!