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

Is it possible to use Torchserve for Huggingface transformers` VisionEncoderDecoderModel? #1902

Closed
kangsan0420 opened this issue Oct 13, 2022 · 1 comment
Assignees
Labels
example triaged Issue has been reviewed and triaged

Comments

@kangsan0420
Copy link

kangsan0420 commented Oct 13, 2022

Hi, I'm wondering whether I can use Torchserve for the below model.
(I'd like to use it for TrOCR)

from transformers import VisionEncoderDecoderModel
encoder_decoder = ('google/vit-base-patch16-224-in21k', 'kykim/bert-kor-base')
model = VisionEncoderDecoderModel.from_encoder_decoder_pretrained(*encoder_decoder)
@msaroufim msaroufim added the triaged Issue has been reviewed and triaged label Oct 13, 2022
@msaroufim
Copy link
Member

msaroufim commented Oct 13, 2022

@agunapal has recently looked at video models, it should be entirely possible - just need to expand on the huggingface custom handler to support this

This recent PR should give you a good idea on how to do this #1904

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
example triaged Issue has been reviewed and triaged
Projects
None yet
Development

No branches or pull requests

3 participants