We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent cf9b093 commit fe4cb35Copy full SHA for fe4cb35
1 file changed
tests/end_to_end/tpu/gemma3/4b/test_gemma3_multimodal_sft.sh
@@ -66,13 +66,14 @@ python3 -m maxtext.inference.decode \
66
model_name=${MODEL_NAME} \
67
load_parameters_path=${BASE_OUTPUT_DIRECTORY}/multimodal/sft/${run_id}/checkpoints/4/items \
68
per_device_batch_size=1 \
69
- run_name=${run_id}} \
+ run_name=${run_id} \
70
max_prefill_predict_length=272 \
71
max_target_length=300 \
72
steps=1 \
73
async_checkpointing=false \
74
scan_layers=false \
75
use_multimodal=true \
76
+ tokenizer_type=huggingface \
77
prompt=\'Describe\ image\ \<start_of_image\>\' \
78
image_path=\'tests/assets/test_image.jpg\' \
79
attention=\'dot_product\'
0 commit comments