Set pipeline_tag

#1
by grazh - opened

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!

agentlans changed pull request status to merged

Sign up or log in to comment