From b9d699e0e8dbfd879e54c0d2c9b1f52c3d4b1270 Mon Sep 17 00:00:00 2001 From: Emma Lien Date: Mon, 14 Sep 2026 09:11:11 +0000 Subject: [PATCH] Remove to_huggingface.py step in LoRA e2e --- tests/end_to_end/tpu/gemma3/4b/test_gemma3_lora.sh | 8 -------- 1 file changed, 8 deletions(-) diff --git a/tests/end_to_end/tpu/gemma3/4b/test_gemma3_lora.sh b/tests/end_to_end/tpu/gemma3/4b/test_gemma3_lora.sh index 257e89302c..3266545050 100755 --- a/tests/end_to_end/tpu/gemma3/4b/test_gemma3_lora.sh +++ b/tests/end_to_end/tpu/gemma3/4b/test_gemma3_lora.sh @@ -68,11 +68,3 @@ python3 -m maxtext.inference.vllm_decode \ use_chat_template=True \ scan_layers=true \ enable_single_controller=${use_pathways} - -# Step 5: Convert the checkpoint from MaxText format to Hugging Face format -python3 -m maxtext.checkpoint_conversion.to_huggingface \ - model_name=${MODEL_NAME} \ - load_parameters_path=${SCANNED_CKPT_PATH} \ - lora.lora_restore_path=${BASE_OUTPUT_DIRECTORY}/lora/${run_id}/checkpoints/5/model_params \ - base_output_directory=${BASE_OUTPUT_DIRECTORY}/to_huggingface/unscanned/${run_id} \ - scan_layers=true