diff --git a/.github/workflows/unit_tests.yml b/.github/workflows/unit_tests.yml index c6a8c8f..0275d6e 100644 --- a/.github/workflows/unit_tests.yml +++ b/.github/workflows/unit_tests.yml @@ -10,6 +10,10 @@ on: env: CUDA_DEVICE_ORDER: PCI_BUS_ID + UV_TORCH_BACKEND: cu130 + TORCH_VERSION: 2.11.0 + PYTHON_VERSION: 3.14t + UV_PYTHON: 3.14t jobs: @@ -45,7 +49,7 @@ jobs: needs: - list-test-files container: - image: 10.0.13.31:5000/nvidia/cuda:128-ubuntu22.04_0206 + image: 10.0.13.31:5000/nvidia/cuda:130-ubuntu24.04_0325 volumes: - /monster/ci/env/entrypoint.sh:/etc/profile.d/01-entrypoint.sh - /monster/ci/models:/monster/data/model @@ -60,7 +64,7 @@ jobs: - name: Print Env run: | - env_name="cu128_torch2.8.0_py3.12_tokenicer_${{ matrix.test_script }}" + env_name="cu130_torch_py3.14_tokenicer_${{ matrix.test_script }}" echo "env name: $env_name" @@ -69,11 +73,11 @@ jobs: - name: install requirements run: | uv pip install parameterized pytest pytest-xdist jinja2 transformers sentencepiece tiktoken protobuf -U - if [[ "${{ matrix.test_script }}" == "test_gemma3.py" || "${{ matrix.test_script }}" == "test_loop_models.py" ]]; then + if [[ "${{ matrix.test_script }}" == "test_gemma3.py" || "${{ matrix.test_script }}" == "test_loop_models.py" || "${{ matrix.test_script }}" == "test_pad_token.py" ]]; then uv pip install torch fi - - name: install + - name: Install run: uv pip install . - name: test