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

[FEATURE] Trace Summarization models to TorchScript and Onnx format #303

Open
dhrubo-os opened this issue Sep 27, 2023 · 4 comments
Open
Labels
CCI enhancement New feature or request good first issue Good for newcomers

Comments

@dhrubo-os
Copy link
Collaborator

We are planning to add more model supports in ML-Commons. Reference issue

The target of this issue is to trace 3 popular pre-trained summarization models to TorchScript and Onnx format. In this repo we traced pre-trained sentence embedding models into torchscript and onnx

We need to build the similar method to trace summarization models. Primarily we can target these models:

  1. facebook/bart-large-cnn
  2. sshleifer/distilbart-cnn-12-6
  3. philschmid/bart-large-cnn-samsum

I created a feature branch : feature/summarization_model_conversation/. All the development of this issue should be done in that branch.

@dhrubo-os dhrubo-os added enhancement New feature or request untriaged good first issue Good for newcomers CCI and removed untriaged labels Sep 27, 2023
@vicradon
Copy link

vicradon commented Oct 6, 2023

I'd like to work on this one

@dhrubo-os
Copy link
Collaborator Author

Sure, assigning it to you.

@dhrubo-os
Copy link
Collaborator Author

Hi @vicradon any update on this issue?

@vicradon
Copy link

Hi @dhrubo-os
Can you reassign this issue to someone else?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CCI enhancement New feature or request good first issue Good for newcomers
Projects
None yet
Development

No branches or pull requests

2 participants