Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Support for ONNX export of UMT5 #2142

Open
cyanic-selkie opened this issue Dec 30, 2024 · 0 comments
Open

Support for ONNX export of UMT5 #2142

cyanic-selkie opened this issue Dec 30, 2024 · 0 comments
Labels
feature-request New feature or request

Comments

@cyanic-selkie
Copy link

Feature request

Add UMT5 support to onnx_export_from_model.

Motivation

Being able to deploy UMT5 models (like pile-t5) to production using onnx.

Your contribution

It seems to me like the T5 config should actually work for UMT5, but I can't get it to work. Given a list of steps I have to do, I'd be happy to submit a PR.
@echarlaix echarlaix added the feature-request New feature or request label Jan 6, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
feature-request New feature or request
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@cyanic-selkie @echarlaix