Merge pull request #3797 from AI-Hypercomputer:aireen/fix_vision_output_length

Google-ML-Automation · Google-ML-Automation · commit 1adb45ce021c · 2026-05-01T15:54:19.000-07:00
PiperOrigin-RevId: 908940648
diff --git a/tests/end_to_end/tpu/gemma4/26b/convert_gemma4.sh b/tests/end_to_end/tpu/gemma4/26b/convert_gemma4.sh
@@ -55,7 +55,6 @@ if [ ${USE_MULTIMODAL} == true ]; then
         dtype=float32 \
         matmul_precision=highest \
         per_device_batch_size=1 \
-        vision_output_length=280 \
         attention=dot_product \
         prompt="${TEST_PROMPT}" \
         image_path=${TEST_IMAGE} \
diff --git a/tests/end_to_end/tpu/gemma4/31b/convert_gemma4.sh b/tests/end_to_end/tpu/gemma4/31b/convert_gemma4.sh
@@ -55,7 +55,6 @@ if [ ${USE_MULTIMODAL} == true ]; then
         dtype=float32 \
         matmul_precision=highest \
         per_device_batch_size=1 \
-        vision_output_length=280 \
         attention=dot_product \
         prompt="${TEST_PROMPT}" \
         image_path=${TEST_IMAGE} \
