diff --git a/.github/workflows/pull.yml b/.github/workflows/pull.yml index 5cc117fb3f7..81948e4e827 100644 --- a/.github/workflows/pull.yml +++ b/.github/workflows/pull.yml @@ -290,6 +290,36 @@ jobs: # run e2e (export, tokenizer and runner) PYTHON_EXECUTABLE=python bash .ci/scripts/test_llava.sh + test-moshi-linux: + name: test-moshi-linux + uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main + permissions: + id-token: write + contents: read + strategy: + fail-fast: false + with: + runner: linux.2xlarge + docker-image: executorch-ubuntu-22.04-clang12 + submodules: 'true' + ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }} + timeout: 90 + script: | + # The generic Linux job chooses to use base env, not the one setup by the image + CONDA_ENV=$(conda env list --json | jq -r ".envs | .[-1]") + conda activate "${CONDA_ENV}" + + PYTHON_EXECUTABLE=python bash .ci/scripts/setup-linux.sh --build-tool "cmake" + + # install Mimi requirements + bash examples/models/moshi/mimi/install_requirements.sh + + # reinstall executorch + bash ./install_executorch.sh + + # run python unittest + python -m unittest examples.models.moshi.mimi.test_mimi + test-quantized-aot-lib-linux: name: test-quantized-aot-lib-linux uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main diff --git a/examples/models/moshi/mimi/install_requirements.sh b/examples/models/moshi/mimi/install_requirements.sh index 2de86466130..ef915ca7eb2 100755 --- a/examples/models/moshi/mimi/install_requirements.sh +++ b/examples/models/moshi/mimi/install_requirements.sh @@ -8,8 +8,7 @@ set -x pip install -U moshi -pip install bitsandbytes +pip install bitsandbytes soundfile # Run llama2/install requirements for torchao deps SCRIPT_DIR=$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd ) - -bash "$SCRIPT_DIR"/../llama/install_requirements.sh +bash "$SCRIPT_DIR"/../../llama/install_requirements.sh diff --git a/examples/models/moshi/mimi/test_mimi.py b/examples/models/moshi/mimi/test_mimi.py index 74662f61807..8160b5df79c 100644 --- a/examples/models/moshi/mimi/test_mimi.py +++ b/examples/models/moshi/mimi/test_mimi.py @@ -2,7 +2,6 @@ import os import random import unittest -import urllib import numpy as np import requests