File tree Expand file tree Collapse file tree 1 file changed +27
-0
lines changed
Expand file tree Collapse file tree 1 file changed +27
-0
lines changed Original file line number Diff line number Diff line change @@ -290,6 +290,33 @@ jobs:
290290 # run e2e (export, tokenizer and runner)
291291 PYTHON_EXECUTABLE=python bash .ci/scripts/test_llava.sh
292292
293+ test-moshi-linux :
294+ name : test-moshi-linux
295+ uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
296+ permissions :
297+ id-token : write
298+ contents : read
299+ strategy :
300+ fail-fast : false
301+ with :
302+ runner : linux.2xlarge
303+ docker-image : executorch-ubuntu-22.04-clang12
304+ submodules : ' true'
305+ ref : ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
306+ timeout : 90
307+ script : |
308+ # The generic Linux job chooses to use base env, not the one setup by the image
309+ CONDA_ENV=$(conda env list --json | jq -r ".envs | .[-1]")
310+ conda activate "${CONDA_ENV}"
311+
312+ PYTHON_EXECUTABLE=python bash .ci/scripts/setup-linux.sh --build-tool "cmake"
313+
314+ # install Llava requirements
315+ bash examples/models/moshi/install_requirements.sh
316+
317+ # run python unittest
318+ python -m unittest examples.models.moshi.mimi.test_mimi
319+
293320 test-quantized-aot-lib-linux :
294321 name : test-quantized-aot-lib-linux
295322 uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
You can’t perform that action at this time.
0 commit comments