Skip to content

Commit 2f917c3

Browse files
guangy10Guang Yang
andauthored
Add latency measurement for text models (#44)
Co-authored-by: Guang Yang <[email protected]>
1 parent bd46d76 commit 2f917c3

File tree

7 files changed

+709
-940
lines changed

7 files changed

+709
-940
lines changed

.github/workflows/test_models.yml

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@ jobs:
2424
- deit
2525
- distilbert
2626
- dit
27+
- efficientnet
2728
- focalnet
2829
- gemma
2930
- gemma2
@@ -35,8 +36,10 @@ jobs:
3536
- pvt
3637
- qwen2
3738
- roberta
39+
- smollm
3840
- swin
3941
- t5
42+
- vit
4043
executorch-version: ['0.4.0', 'nightly']
4144
python-version: ['3.10', '3.11', '3.12']
4245
os: [macos-15]
@@ -55,7 +58,7 @@ jobs:
5558
- name: Install dependencies for ExecuTorch
5659
run: |
5760
if [ "${{ matrix.executorch-version }}" == "nightly" ]; then
58-
pip install executorch==0.7.0.dev20250330 --extra-index-url "https://download.pytorch.org/whl/nightly/cpu"
61+
pip install executorch==0.7.0.dev20250403 --extra-index-url "https://download.pytorch.org/whl/nightly/cpu"
5962
else
6063
pip install executorch==${{ matrix.executorch-version }}
6164
fi

README.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -152,6 +152,7 @@ We currently support a wide range of popular transformer models, including encod
152152
- [Cvt](https://huggingface.co/microsoft/cvt-13): Convolutional Vision Transformer
153153
- [Deit](https://huggingface.co/facebook/deit-base-distilled-patch16-224): Distilled Data-efficient Image Transformer (base-sized)
154154
- [Dit](https://huggingface.co/microsoft/dit-base-finetuned-rvlcdip): Document Image Transformer (base-sized)
155+
- [EfficientNet](https://huggingface.co/google/efficientnet-b0): EfficientNet (b0-b7 sized)
155156
- [Focalnet](https://huggingface.co/microsoft/focalnet-tiny): FocalNet (tiny-sized)
156157
- [Mobilevit](https://huggingface.co/apple/mobilevit-xx-small): Apple's MobileViT xx-small
157158
- [Mobilevit2](https://huggingface.co/apple/mobilevitv2-1.0-imagenet1k-256): Apple's MobileViTv2

0 commit comments

Comments
 (0)