Skip to content

Commit fe4cb35

Browse files
committed
Fix Gemma-3 4B Multimodal SFT test script Step 4 decode inference tokenizer and run_name
1 parent cf9b093 commit fe4cb35

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

tests/end_to_end/tpu/gemma3/4b/test_gemma3_multimodal_sft.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,13 +66,14 @@ python3 -m maxtext.inference.decode \
6666
model_name=${MODEL_NAME} \
6767
load_parameters_path=${BASE_OUTPUT_DIRECTORY}/multimodal/sft/${run_id}/checkpoints/4/items \
6868
per_device_batch_size=1 \
69-
run_name=${run_id}} \
69+
run_name=${run_id} \
7070
max_prefill_predict_length=272 \
7171
max_target_length=300 \
7272
steps=1 \
7373
async_checkpointing=false \
7474
scan_layers=false \
7575
use_multimodal=true \
76+
tokenizer_type=huggingface \
7677
prompt=\'Describe\ image\ \<start_of_image\>\' \
7778
image_path=\'tests/assets/test_image.jpg\' \
7879
attention=\'dot_product\'

0 commit comments

Comments
 (0)