6767 CUDA_VERSION : 130
6868 UV_TORCH_BACKEND : cu130
6969 TORCH_VERSION : 2.11.0
70- # vllm doesn't support 3.14
71- PYTHON_VERSION : 3.13
72- UV_PYTHON : 3.13
70+ PYTHON_VERSION : 3.14t
71+ UV_PYTHON : 3.14t
7372 # PYTHON_GIL: 0 // test libs don't support yet
7473 IGNORED_TEST_FILES : " test_tgi.py,test_gptneox.py,models/test_mixtral.py,models/test_phi_3_moe.py,test_bits_new.py,models/test_internlm.py,models/test_internlm2_5.py,models/test_xverse.py"
7574 MODEL_TEST_MODE : ${{ github.event.inputs['model-test-mode'] || 'fast' }}
@@ -247,19 +246,19 @@ jobs:
247246 - name : Test pypi pip
248247 env_name : pypi_pip_env_${{ github.run_id }}_${{ github.run_attempt }}
249248 use_pip : ' true'
250- install_cmd : pip install gptqmodel torch -U
249+ install_cmd : pip install gptqmodel torch -U -vvv
251250 - name : Test pypi uv
252251 env_name : pypi_uv_env_${{ github.run_id }}_${{ github.run_attempt }}
253252 use_pip : ' false'
254- install_cmd : uv pip install gptqmodel torch -U
253+ install_cmd : uv pip install gptqmodel torch -U -vvv
255254 - name : Test local pip
256255 env_name : local_pip_env_${{ github.run_id }}_${{ github.run_attempt }}
257256 use_pip : ' true'
258- install_cmd : pip install . torch -U
257+ install_cmd : pip install . torch -U -vvv
259258 - name : Test local uv
260259 env_name : local_uv_env_${{ github.run_id }}_${{ github.run_attempt }}
261260 use_pip : ' false'
262- install_cmd : uv pip install . torch -U
261+ install_cmd : uv pip install . torch -U -vvv
263262 steps :
264263 - name : Checkout Codes
265264 uses : actions/checkout@v6
@@ -278,6 +277,13 @@ jobs:
278277 if [[ "${{ matrix.use_pip }}" == "true" ]]; then
279278 uv pip install pip -U
280279 fi
280+ if python -VV 2>&1 | grep -q '3\.14.*free-threading'; then
281+ echo "[INFO] Python 3.14t detected, installing torchao..."
282+ uv pip install http://10.0.13.31/files/torchao-0.18.0+git13cd013d6-cp314-cp314t-linux_x86_64.whl
283+ else
284+ echo "[INFO] Current Python is not 3.14t, skip torchao."
285+ fi
286+
281287 ${{ matrix.install_cmd }}
282288
283289
0 commit comments