demucs-onnx/.github/workflows/ci.yml at 071a15d7a9c25efa37b896b7ac5a38f0de7637d4 · hi-ogawa/demucs-onnx · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
name: CI

on:
  push:
    branches:
      - main
  pull_request:

concurrency:
  group: ci-${{ github.event.pull_request.number || github.ref }}
  cancel-in-progress: true

permissions:
  contents: read

jobs:
  test:
    runs-on: ubuntu-latest
    timeout-minutes: 60
    env:
      # Limit model export so it does not saturate the CI runner.
      # https://docs.pytorch.org/docs/stable/threading_environment_variables.html
      OMP_NUM_THREADS: 2

    steps:
      - uses: actions/checkout@v6

      - uses: actions/setup-node@v6
        with:
          node-version: "lts/*"

      - run: corepack enable

      - uses: astral-sh/setup-uv@v8.3.2
        with:
          enable-cache: true

      - uses: dtolnay/rust-toolchain@stable

      - uses: taiki-e/install-action@v2.83.1
        with:
          tool: wasm-pack@0.15.0

      - name: Cache model export inputs
        uses: actions/cache@v6.1.0
        with:
          path: |
            ~/.cache/torch/hub/checkpoints
            data/onnx
          key: ${{ runner.os }}-htdemucs-${{ hashFiles('uv.lock', 'tools/model-export/build_models.py', 'tools/model-export/export_onnx.py') }}

      - run: pnpm install --frozen-lockfile

      - run: pnpm build:model htdemucs

      - run: |
          pnpm cli-separate \
            --name htdemucs \
            --two-stems bass \
            --method minus \
            fixtures/sine-10s.wav data/ci-output
          test -s data/ci-output/bass.wav
          test -s data/ci-output/no_bass.wav

      - run: pnpm build:wasm

      - run: pnpm lint-check

      - run: pnpm -C packages/app exec playwright install chromium --with-deps --no-shell

      - run: pnpm test-e2e