Skip to content

Commit 14bf19e

Browse files
authored
[TPU][TEST] Fix the downloading issue in TPU v1 test 11. (#21418)
Signed-off-by: Qiliang Cui <derrhein@gmail.com>
1 parent 4ac7713 commit 14bf19e

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

.buildkite/scripts/hardware_ci/run-tpu-v1-test.sh

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,8 @@ echo "Results will be stored in: $RESULTS_DIR"
6262
echo "--- Installing Python dependencies ---"
6363
python3 -m pip install --progress-bar off git+https://github.com/thuml/depyf.git \
6464
&& python3 -m pip install --progress-bar off pytest pytest-asyncio tpu-info \
65-
&& python3 -m pip install --progress-bar off lm_eval[api]==0.4.4
65+
&& python3 -m pip install --progress-bar off lm_eval[api]==0.4.4 \
66+
&& python3 -m pip install --progress-bar off hf-transfer
6667
echo "--- Python dependencies installed ---"
6768
export VLLM_USE_V1=1
6869
export VLLM_XLA_CHECK_RECOMPILATION=1
@@ -150,7 +151,7 @@ run_and_track_test 9 "test_multimodal.py" \
150151
run_and_track_test 10 "test_pallas.py" \
151152
"python3 -m pytest -s -v /workspace/vllm/tests/v1/tpu/test_pallas.py"
152153
run_and_track_test 11 "test_struct_output_generate.py" \
153-
"python3 -m pytest -s -v /workspace/vllm/tests/v1/entrypoints/llm/test_struct_output_generate.py -k \"not test_structured_output_with_reasoning_matrices\""
154+
"HF_HUB_DISABLE_XET=1 python3 -m pytest -s -v /workspace/vllm/tests/v1/entrypoints/llm/test_struct_output_generate.py -k \"not test_structured_output_with_reasoning_matrices\""
154155
run_and_track_test 12 "test_moe_pallas.py" \
155156
"python3 -m pytest -s -v /workspace/vllm/tests/tpu/test_moe_pallas.py"
156157
run_and_track_test 13 "test_lora.py" \

0 commit comments

Comments
 (0)