Skip to content

Commit 9c6ebc3

Browse files
committed
Fix test_model.sh for vision text decoder
1 parent eccc633 commit 9c6ebc3

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

.ci/scripts/test_model.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -87,8 +87,8 @@ test_model() {
8787
bash examples/models/llava/install_requirements.sh
8888
STRICT="--no-strict"
8989
fi
90-
if [[ "$MODEL_NAME" == "llama3_2_vision_encoder" ]]; then
91-
# Install requirements for llama vision
90+
if [[ "$MODEL_NAME" == "llama3_2_vision_encoder" || "$MODEL_NAME" == "llama3_2_text_decoder" ]]; then
91+
# Install requirements for llama vision.
9292
bash examples/models/llama3_2_vision/install_requirements.sh
9393
fi
9494
# python3 -m examples.portable.scripts.export --model_name="llama2" should works too

0 commit comments

Comments
 (0)