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.

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

Add support for EncoderDecoderModel #462

Closed
alvations opened this issue Nov 10, 2022 · 1 comment
Closed

Add support for EncoderDecoderModel #462

alvations opened this issue Nov 10, 2022 · 1 comment
Assignees

Comments

@alvations
Copy link

Feature request

There's already support for marian and various LLMs. But sometimes users create their own generic EncoderDecoderModel, e.g.

from transformers import EncoderDecoderModel
from optimum.onnxruntime import ORTModelForSeq2SeqLM

model = EncoderDecoderModel.from_encoder_decoder_pretrained("bert-base-multilingual-cased", "bert-base-multilingual-cased") 

model.save_pretrained("model_dir")

# Should be able to load this, but isn't supported yet.
ort_model = ORTModelForSeq2SeqLM.from_pretrained("model_dir", from_transformers=True)

Motivation

The EncoderDecoderModel is generic enough to cover quite a lot of use-cases but this is might be hard too since it can most probably only cover EncoderDecoder of ORT supported LLMs

Your contribution

Maybe, if there's some guidance on how to do so.

@mht-sharma mht-sharma self-assigned this Sep 1, 2023
@mht-sharma
Copy link
Contributor

Fixed in #851

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants