Sept. 22, 2022, 9:17 p.m. | /u/Lost-Letterhead2105

Machine Learning www.reddit.com

Hi all

I am working on a project where I fine-tuned a Pegasus model on the Reddit dataset. Now, I need to convert the [fine-tuned model](https://huggingface.co/QuickRead/pegasus-reddit-7e05-new) to ONNX for the deployment stage. I have followed this guide from Huggingface to [convert to the ONNX model for unsupported architect](https://huggingface.co/docs/transformers/serialization#implementing-a-custom-onnx-configuration)s. I got it done but the ONNX model can't generate text. Turned out that Pegasus is an encoder-decoder model and most guides are for either encoder-model (e.g. BERT) or decoder-model (e.g. GPT2). …

machinelearning onnx

Data Scientist (m/f/x/d)

@ Symanto Research GmbH & Co. KG | Spain, Germany

Automated Greenhouse Expert - Phenotyping & Data Analysis (all genders)

@ Bayer | Frankfurt a.M., Hessen, DE

Machine Learning Scientist II

@ Expedia Group | India - Bengaluru

Data Engineer/Senior Data Engineer, Bioinformatics

@ Flagship Pioneering, Inc. | Cambridge, MA USA

Intern (AI lab)

@ UL Solutions | Dublin, Co. Dublin, Ireland

Senior Operations Research Analyst / Predictive Modeler

@ LinQuest | Colorado Springs, Colorado, United States