Skip to content

Commit 328c72c

Browse files
committed
Remove future implementation
1 parent a6f96a2 commit 328c72c

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

examples/models/llama2/export_llama_lib.py

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

8080

8181
EXECUTORCH_DEFINED_MODELS = ["llama2", "llama3", "llama3_1", "llama3_2"]
82-
TORCHTUNE_DEFINED_MODELS = ["llama3_2_vision"]
82+
TORCHTUNE_DEFINED_MODELS = []
8383

8484

8585
class WeightType(Enum):
@@ -800,8 +800,7 @@ def _load_llama_model(
800800
modelname = "llama2"
801801
model_class_name = "Llama2Model"
802802
elif modelname in TORCHTUNE_DEFINED_MODELS:
803-
if modelname == "llama3_2_vision":
804-
model_class_name = "Llama3_2Decoder"
803+
raise NotImplementedError("Torchtune Llama models are not yet supported in ExecuTorch export.")
805804
else:
806805
raise ValueError(f"{modelname} is not a valid Llama model.")
807806

0 commit comments

Comments
 (0)