mirror of
https://github.com/multimodal-art-projection/YuE.git
synced 2026-10-21 06:30:25 +08:00
38 lines
898 B
YAML
38 lines
898 B
YAML
name: CPU tests
|
|
|
|
on:
|
|
push:
|
|
pull_request:
|
|
|
|
permissions:
|
|
contents: read
|
|
|
|
jobs:
|
|
test:
|
|
runs-on: ubuntu-latest
|
|
env:
|
|
CUDA_VISIBLE_DEVICES: ""
|
|
HF_HUB_OFFLINE: "1"
|
|
TRANSFORMERS_OFFLINE: "1"
|
|
OMP_NUM_THREADS: "2"
|
|
MKL_NUM_THREADS: "2"
|
|
steps:
|
|
- uses: actions/checkout@v4
|
|
- uses: actions/setup-python@v5
|
|
with:
|
|
python-version: "3.12"
|
|
cache: pip
|
|
- name: Install CPU runtime and tests
|
|
run: |
|
|
python -m pip install --upgrade pip
|
|
python -m pip install torch==2.10.0 --index-url https://download.pytorch.org/whl/cpu
|
|
python -m pip install '.[test]'
|
|
- name: Check public commands
|
|
run: |
|
|
yue2 --help
|
|
yue2 generate --help
|
|
yue2 batch --help
|
|
yue2 doctor
|
|
- name: Run tests without model downloads
|
|
run: python -m pytest -q
|