From 22056215a609e513d6ef7718136cb2302b3d14ff Mon Sep 17 00:00:00 2001 From: CSY-ModelCloud Date: Mon, 30 Mar 2026 10:05:11 +0800 Subject: [PATCH 1/2] [CI] use latest docker & env --- .github/workflows/unit_tests.yml | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/.github/workflows/unit_tests.yml b/.github/workflows/unit_tests.yml index c6a8c8f..e2885a9 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" @@ -73,7 +77,7 @@ jobs: uv pip install torch fi - - name: install + - name: Install run: uv pip install . - name: test From 62a00084162b270e1b837e17862869562f582a5e Mon Sep 17 00:00:00 2001 From: CSY-ModelCloud Date: Mon, 30 Mar 2026 10:21:14 +0800 Subject: [PATCH 2/2] [CI] install torch for test_pad_token --- .github/workflows/unit_tests.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/unit_tests.yml b/.github/workflows/unit_tests.yml index e2885a9..0275d6e 100644 --- a/.github/workflows/unit_tests.yml +++ b/.github/workflows/unit_tests.yml @@ -73,7 +73,7 @@ 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