Skip to content

Commit c3b6c58

Browse files
author
Guang Yang
committed
Enable t5 and similar models
1 parent 2b65e30 commit c3b6c58

29 files changed

+1839
-314
lines changed

.github/workflows/test_export.yml

Lines changed: 0 additions & 35 deletions
This file was deleted.

.github/workflows/test_models.yml

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,12 +18,18 @@ jobs:
1818
python-version: ['3.10', '3.11', '3.12']
1919
os: [macos-15]
2020
test-modeling:
21-
- test_modeling_gemma2.py
21+
- test_modeling.py
22+
- test_modeling_albert.py
23+
- test_modeling_bert.py
24+
- test_modeling_distilbert.py
25+
- test_modeling_enrobert.py
2226
- test_modeling_gemma.py
27+
- test_modeling_gemma2.py
2328
- test_modeling_llama.py
2429
- test_modeling_olmo.py
25-
- test_modeling.py
2630
- test_modeling_qwen2.py
31+
- test_modeling_roberta.py
32+
- test_modeling_t5.py
2733

2834
runs-on: ${{ matrix.os }}
2935
steps:

optimum/executorch/__init__.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,11 +20,17 @@
2020
_import_structure = {
2121
"modeling": [
2222
"ExecuTorchModelForCausalLM",
23+
"ExecuTorchModelForMaskedLM",
24+
"ExecuTorchModelForSeq2SeqLM",
2325
],
2426
}
2527

2628
if TYPE_CHECKING:
27-
from .modeling import ExecuTorchModelForCausalLM
29+
from .modeling import (
30+
ExecuTorchModelForCausalLM,
31+
ExecuTorchModelForMaskedLM,
32+
ExecuTorchModelForSeq2SeqLM,
33+
)
2834
else:
2935
import sys
3036

0 commit comments

Comments
 (0)