Skip to content

Add new GA models (next set) #12398

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

Draft
wants to merge 2 commits into
base: main
Choose a base branch
from
Draft

Add new GA models (next set) #12398

wants to merge 2 commits into from

Conversation

mergennachin
Copy link
Contributor

@mergennachin mergennachin commented Jul 11, 2025

Add next set of GA models

Summary:

On top of #12340

Add a few more tasks:

  1. Image-Text Understanding (OpenCLIP)
  2. Semantic Text Search (Sentence Transformers)
  3. Document Q&A (DistilBERT QA)
  4. Super resolution (Real-ESRGAN)
  5. Audio Classification (AST)
  6. Text Sentiment Analysis (RoBERTa)
  7. Depth estimation (Depth Anything 2)

Copy link

pytorch-bot bot commented Jul 11, 2025

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/12398

Note: Links to docs will display an error until the docs builds have been completed.

❌ 1 New Failure, 1 Cancelled Job, 7 Unrelated Failures

As of commit 0ad47d3 with merge base 7c300e7 (image):

NEW FAILURE - The following job has failed:

CANCELLED JOB - The following job was cancelled. Please retry:

FLAKY - The following job failed but was likely due to flakiness present on trunk:

BROKEN TRUNK - The following jobs failed but was present on the merge base:

👉 Rebase onto the `viable/strict` branch to avoid these failures

This comment was automatically generated by Dr. CI and updates every 15 minutes.

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Jul 11, 2025
Copy link

This PR needs a release notes: label

If your change should be included in the release notes (i.e. would users of this library care about this change?), please use a label starting with release notes:. This helps us keep track and include your important work in the next release notes.

To add a label, you can comment to pytorchbot, for example
@pytorchbot label "release notes: none"

For more information, see
https://github.com/pytorch/pytorch/wiki/PyTorch-AutoLabel-Bot#why-categorize-for-release-notes-and-how-does-it-work.

@mergennachin mergennachin changed the title Add new GA models (final) Add new GA models (next set) Jul 11, 2025
@mergennachin mergennachin force-pushed the add_new_models_2 branch 2 times, most recently from fbd8064 to 87a2817 Compare July 11, 2025 19:25
Added 8 new representative models to ExecutorTorch examples:

EfficientNet-B4: Image classification with CNN architecture
DETR-ResNet50: Object detection using transformer decoder
SegFormer-ADE: Semantic segmentation transformer
Swin2SR: Super-resolution with Swin transformer
ALBERT: Lightweight BERT for NLP tasks
TrOCR: Optical character recognition transformer
Wav2Vec2: Cross-lingual speech representation learning
All models include XNNPACK backend support with appropriate quantization configurations and full CI integration.

Test plan:

Validate model export and execution with portable backend
Test XNNPACK delegation and quantization (with appropriate exclusions)
Integrate into CI workflows for automated testing
Verify all models perform their intended tasks accurately
Summary:

Add a few more tasks:

1. Image-Text Understanding (OpenCLIP)
2. Semantic Text Search (Sentence Transformers)
3. Document Q&A (DistilBERT QA)
4. Practical Image Enhancement (Real-ESRGAN)
5. Audio Classification (AST)
6. Text Sentiment Analysis (RoBERTa)
7. Depth estimation (Depth Anything 2)
@digantdesai
Copy link
Contributor

@mergennachin - As I was going through this list, I realized most of them are from HF transformers. So,
(1) if we should continue adding more HF transformer based models on optimum-et, instead of here.
(2) As I looked on optimum-et[1], I saw many already there i.e. Albert, DistillBert, EfficientNet, Roberta, Swin. Even if we don't do (1) - we should deduplicate unless you have preference for keeping them in ET CI for some reason.

[1] https://github.com/huggingface/optimum-executorch/tree/main/tests/models

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants