Skip to content

Commit 66d357a

Browse files
committed
fix gemma3 to_maxtext flag
1 parent be7748b commit 66d357a

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

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

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,7 @@ python3 -m maxtext.checkpoint_conversion.to_maxtext \
3939
scan_layers=false \
4040
hardware=cpu skip_jax_distributed_system=True \
4141
checkpoint_storage_use_zarr3=False checkpoint_storage_use_ocdbt=False \
42+
--lazy_load_tensors=False \
4243
--eager_load_method='transformers'
4344

4445
UNSCANNED_CKPT_PATH=${BASE_OUTPUT_DIRECTORY}/unscanned/${run_id}/0/items
@@ -52,6 +53,7 @@ python3 -m maxtext.checkpoint_conversion.to_maxtext \
5253
scan_layers=true \
5354
hardware=cpu skip_jax_distributed_system=True \
5455
checkpoint_storage_use_zarr3=False checkpoint_storage_use_ocdbt=False \
56+
--lazy_load_tensors=False \
5557
--eager_load_method='transformers'
5658

5759
SCANNED_CKPT_PATH=${BASE_OUTPUT_DIRECTORY}/scanned/${run_id}/0/items

0 commit comments

Comments
 (0)