Skip to content

Commit 850e7c8

Browse files
committed
Revert "Remove future implementation"
This reverts commit 60a21c1.
1 parent c7338da commit 850e7c8

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

examples/models/llama2/export_llama_lib.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@
7676

7777

7878
EXECUTORCH_DEFINED_MODELS = ["llama2", "llama3", "llama3_1", "llama3_2"]
79-
TORCHTUNE_DEFINED_MODELS = []
79+
TORCHTUNE_DEFINED_MODELS = ["llama3_2_vision"]
8080

8181

8282
class WeightType(Enum):
@@ -798,7 +798,8 @@ def _load_llama_model(
798798
modelname = "llama2"
799799
model_class_name = "Llama2Model"
800800
elif modelname in TORCHTUNE_DEFINED_MODELS:
801-
raise NotImplementedError("Torchtune Llama models are not yet supported in ExecuTorch export.")
801+
if modelname == "llama3_2_vision":
802+
model_class_name = "Llama3_2Decoder"
802803
else:
803804
raise ValueError(f"{modelname} is not a valid Llama model.")
804805

0 commit comments

Comments
 (0)