Update and disable push_tests_mps workflow
#15
Workflow file for this run
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: Fast mps tests on main | |
| on: | |
| pull_request: | |
| branches: | |
| - main | |
| paths: | |
| - "src/diffusers/**.py" | |
| - "benchmarks/**.py" | |
| - "examples/**.py" | |
| - "scripts/**.py" | |
| - "tests/**.py" | |
| - ".github/**.yml" | |
| - "utils/**.py" | |
| env: | |
| DIFFUSERS_IS_CI: yes | |
| OMP_NUM_THREADS: 8 | |
| MKL_NUM_THREADS: 8 | |
| HF_HUB_ENABLE_HF_TRANSFER: 1 | |
| PYTEST_TIMEOUT: 600 | |
| RUN_SLOW: no | |
| HF_TOKEN: ${{ secrets.HF_TOKEN }} | |
| PYTORCH_MPS_HIGH_WATERMARK_RATIO: 0.0 | |
| concurrency: | |
| group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }} | |
| cancel-in-progress: true | |
| jobs: | |
| run_fast_tests_apple_m1: | |
| strategy: | |
| fail-fast: false | |
| matrix: | |
| config: | |
| # - name: Fast Pipelines MPS tests | |
| # framework: pytorch_pipelines | |
| # runner: macos-13-xlarge | |
| # report: torch_mps_pipelines | |
| # - name: Fast Models MPS tests | |
| # framework: pytorch_models | |
| # runner: macos-13-xlarge | |
| # report: torch_mps_models | |
| - name: Fast Schedulers MPS tests | |
| framework: pytorch_schedulers | |
| runner: macos-13-xlarge | |
| report: torch_mps_schedulers | |
| # - name: Fast Others MPS tests | |
| # framework: pytorch_others | |
| # runner: macos-13-xlarge | |
| # report: torch_mps_others | |
| # - name: Fast Single File MPS tests | |
| # framework: pytorch_single_file | |
| # runner: macos-13-xlarge | |
| # report: torch_mps_single_file | |
| # - name: Fast Lora MPS tests | |
| # framework: pytorch_lora | |
| # runner: macos-13-xlarge | |
| # report: torch_mps_lora | |
| # - name: Fast Quantization MPS tests | |
| # framework: pytorch_quantization | |
| # runner: macos-13-xlarge | |
| # report: torch_mps_quantization | |
| name: ${{ matrix.config.name }} | |
| runs-on: ${{ matrix.config.runner }} | |
| defaults: | |
| run: | |
| shell: arch -arch arm64 bash {0} | |
| steps: | |
| - name: Checkout diffusers | |
| uses: actions/checkout@v3 | |
| with: | |
| fetch-depth: 2 | |
| - name: Clean checkout | |
| run: | | |
| git clean -fxd | |
| - name: Setup miniconda | |
| uses: ./.github/actions/setup-miniconda | |
| with: | |
| python-version: 3.9 | |
| - name: Install dependencies | |
| run: | | |
| ${CONDA_RUN} python -m pip install --upgrade pip uv | |
| ${CONDA_RUN} python -m uv pip install -e ".[quality,test]" | |
| ${CONDA_RUN} python -m uv pip install hf_transfer | |
| ${CONDA_RUN} python -m uv pip install peft | |
| ${CONDA_RUN} python -m uv pip install torch torchvision torchaudio | |
| ${CONDA_RUN} python -m uv pip install accelerate@git+https://github.com/huggingface/accelerate.git | |
| ${CONDA_RUN} python -m uv pip install transformers --upgrade | |
| - name: Environment | |
| shell: arch -arch arm64 bash {0} | |
| run: | | |
| ${CONDA_RUN} python utils/print_env.py | |
| - name: Run fast PyTorch Pipeline MPS tests | |
| if: ${{ matrix.config.framework == 'pytorch_pipelines' }} | |
| run: | | |
| ${CONDA_RUN} python -m pytest -n 0 -s -v -k "not Flax and not Onnx" \ | |
| --make-reports=tests_${{ matrix.config.report }} \ | |
| tests/pipelines | |
| - name: Run fast PyTorch Models MPS tests | |
| if: ${{ matrix.config.framework == 'pytorch_models' }} | |
| run: | | |
| ${CONDA_RUN} python -m pytest -n 0 -s -v -k "not Flax and not Onnx and not Dependency" \ | |
| --make-reports=tests_${{ matrix.config.report }} \ | |
| tests/models/ | |
| - name: Run fast PyTorch Schedulers MPS tests | |
| if: ${{ matrix.config.framework == 'pytorch_schedulers' }} | |
| run: | | |
| ${CONDA_RUN} python -m pytest -n 0 -s -v -k "not Flax and not Onnx and not Dependency" \ | |
| --make-reports=tests_${{ matrix.config.report }} \ | |
| tests/schedulers/ | |
| - name: Run fast PyTorch Others MPS tests | |
| if: ${{ matrix.config.framework == 'pytorch_others' }} | |
| run: | | |
| ${CONDA_RUN} python -m pytest -n 0 -s -v \ | |
| --make-reports=tests_${{ matrix.config.report }} \ | |
| tests/others/ | |
| - name: Run fast PyTorch Single File MPS tests | |
| if: ${{ matrix.config.framework == 'pytorch_single_file' }} | |
| run: | | |
| ${CONDA_RUN} python -m pytest -n 0 -s -v \ | |
| --make-reports=tests_${{ matrix.config.report }} \ | |
| tests/single_file/ | |
| - name: Run fast PyTorch Lora MPS tests | |
| if: ${{ matrix.config.framework == 'pytorch_lora' }} | |
| run: | | |
| ${CONDA_RUN} python -m pytest -n 0 -s -v \ | |
| --make-reports=tests_${{ matrix.config.report }} \ | |
| tests/lora/ | |
| # - name: Run fast PyTorch Quantization MPS tests | |
| # if: ${{ matrix.config.framework == 'pytorch_quantization' }} | |
| # env: | |
| # HF_TOKEN: ${{ secrets.HF_TOKEN }} | |
| # run: | | |
| # ${CONDA_RUN} python -m pytest -n 0 -s -v \ | |
| # --make-reports=tests_${{ matrix.config.report }} \ | |
| # tests/quantization/ | |
| - name: Failure short reports | |
| if: ${{ failure() }} | |
| run: cat reports/tests_${{ matrix.config.report }}_failures_short.txt | |
| - name: Test suite reports artifacts | |
| if: ${{ always() }} | |
| uses: actions/upload-artifact@v4 | |
| with: | |
| name: pr_${{ matrix.config.framework }}_${{ matrix.config.report }}_test_reports | |
| path: reports |