Skip to content

Commit 038265e

Browse files
authored
Skip llama3_2_text_decoder because it takes too long to export
1 parent 1df29d1 commit 038265e

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

examples/models/__init__.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,8 @@
1919
"llama2": ("llama", "Llama2Model"),
2020
"llama": ("llama", "Llama2Model"),
2121
"llama3_2_vision_encoder": ("llama3_2_vision", "FlamingoVisionEncoderModel"),
22-
"llama3_2_text_decoder": ("llama3_2_vision", "Llama3_2Decoder"),
22+
# TODO: This take too long to export on both Linux and MacOS (> 6 hours)
23+
# "llama3_2_text_decoder": ("llama3_2_vision", "Llama3_2Decoder"),
2324
"lstm": ("lstm", "LSTMModel"),
2425
"mobilebert": ("mobilebert", "MobileBertModelExample"),
2526
"mv2": ("mobilenet_v2", "MV2Model"),

0 commit comments

Comments
 (0)