Skip to content

Commit 62ccef1

Browse files
committed
Add qwen to export_llama --models
1 parent dadf004 commit 62ccef1

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

examples/models/llama/export_llama_lib.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -84,13 +84,15 @@
8484
verbosity_setting = None
8585

8686

87+
# All models that leverage the transformer architecture defined in llama_transformer.py.
8788
EXECUTORCH_DEFINED_MODELS = [
8889
"stories110m",
8990
"llama2",
9091
"llama3",
9192
"llama3_1",
9293
"llama3_2",
9394
"static_llama",
95+
"qwen2_5",
9496
]
9597
TORCHTUNE_DEFINED_MODELS = ["llama3_2_vision"]
9698

0 commit comments

Comments
 (0)