Skip to content

[CI] use latest docker & env & fix test #56

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
Qubitium merged 2 commits into from
Mar 30, 2026
Merged

[CI] use latest docker & env & fix test #56

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
2205621
[CI] use latest docker & env
CSY-ModelCloud Mar 30, 2026
62a0008
[CI] install torch for test_pad_token
CSY-ModelCloud Mar 30, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 8 additions & 4 deletions .github/workflows/unit_tests.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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:

Expand Down Expand Up @@ -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
Expand All @@ -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"

Expand All @@ -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
Expand Down
Loading