Skip to content

Commit b18f98e

Browse files
sayakpaulsayakpaul
committed
disable onnx, mps, flax
1 parent 80f27d7 commit b18f98e

File tree

7 files changed

+334
-339
lines changed

7 files changed

+334
-339
lines changed

‎.github/workflows/build_docker_images.yml‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -75,10 +75,10 @@ jobs:
7575
- diffusers-pytorch-cuda
7676
- diffusers-pytorch-xformers-cuda
7777
- diffusers-pytorch-minimum-cuda
78-
- diffusers-flax-cpu
79-
- diffusers-flax-tpu
80-
- diffusers-onnxruntime-cpu
81-
- diffusers-onnxruntime-cuda
78+
# - diffusers-flax-cpu
79+
# - diffusers-flax-tpu
80+
# - diffusers-onnxruntime-cpu
81+
# - diffusers-onnxruntime-cuda
8282
- diffusers-doc-builder
8383

8484
steps:

‎.github/workflows/nightly_tests.yml‎

Lines changed: 100 additions & 100 deletions
Original file line numberDiff line numberDiff line change
@@ -321,54 +321,54 @@ jobs:
321321
name: torch_minimum_version_cuda_test_reports
322322
path: reports
323323

324-
run_nightly_onnx_tests:
325-
name: Nightly ONNXRuntime CUDA tests on Ubuntu
326-
runs-on:
327-
group: aws-g4dn-2xlarge
328-
container:
329-
image: diffusers/diffusers-onnxruntime-cuda
330-
options: --gpus 0 --shm-size "16gb" --ipc host
331-
332-
steps:
333-
- name: Checkout diffusers
334-
uses: actions/checkout@v3
335-
with:
336-
fetch-depth: 2
337-
338-
- name: NVIDIA-SMI
339-
run: nvidia-smi
340-
341-
- name: Install dependencies
342-
run: |
343-
python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH"
344-
python -m uv pip install -e [quality,test]
345-
pip uninstall accelerate -y && python -m uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git
346-
python -m uv pip install pytest-reportlog
347-
- name: Environment
348-
run: python utils/print_env.py
349-
350-
- name: Run Nightly ONNXRuntime CUDA tests
351-
env:
352-
HF_TOKEN: ${{ secrets.DIFFUSERS_HF_HUB_READ_TOKEN }}
353-
run: |
354-
python -m pytest -n 1 --max-worker-restart=0 --dist=loadfile \
355-
-s -v -k "Onnx" \
356-
--make-reports=tests_onnx_cuda \
357-
--report-log=tests_onnx_cuda.log \
358-
tests/
359-
360-
- name: Failure short reports
361-
if: ${{ failure() }}
362-
run: |
363-
cat reports/tests_onnx_cuda_stats.txt
364-
cat reports/tests_onnx_cuda_failures_short.txt
365-
366-
- name: Test suite reports artifacts
367-
if: ${{ always() }}
368-
uses: actions/upload-artifact@v4
369-
with:
370-
name: tests_onnx_cuda_reports
371-
path: reports
324+
# run_nightly_onnx_tests:
325+
# name: Nightly ONNXRuntime CUDA tests on Ubuntu
326+
# runs-on:
327+
# group: aws-g4dn-2xlarge
328+
# container:
329+
# image: diffusers/diffusers-onnxruntime-cuda
330+
# options: --gpus 0 --shm-size "16gb" --ipc host
331+
332+
# steps:
333+
# - name: Checkout diffusers
334+
# uses: actions/checkout@v3
335+
# with:
336+
# fetch-depth: 2
337+
338+
# - name: NVIDIA-SMI
339+
# run: nvidia-smi
340+
341+
# - name: Install dependencies
342+
# run: |
343+
# python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH"
344+
# python -m uv pip install -e [quality,test]
345+
# pip uninstall accelerate -y && python -m uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git
346+
# python -m uv pip install pytest-reportlog
347+
# - name: Environment
348+
# run: python utils/print_env.py
349+
350+
# - name: Run Nightly ONNXRuntime CUDA tests
351+
# env:
352+
# HF_TOKEN: ${{ secrets.DIFFUSERS_HF_HUB_READ_TOKEN }}
353+
# run: |
354+
# python -m pytest -n 1 --max-worker-restart=0 --dist=loadfile \
355+
# -s -v -k "Onnx" \
356+
# --make-reports=tests_onnx_cuda \
357+
# --report-log=tests_onnx_cuda.log \
358+
# tests/
359+
360+
# - name: Failure short reports
361+
# if: ${{ failure() }}
362+
# run: |
363+
# cat reports/tests_onnx_cuda_stats.txt
364+
# cat reports/tests_onnx_cuda_failures_short.txt
365+
366+
# - name: Test suite reports artifacts
367+
# if: ${{ always() }}
368+
# uses: actions/upload-artifact@v4
369+
# with:
370+
# name: tests_onnx_cuda_reports
371+
# path: reports
372372

373373
run_nightly_quantization_tests:
374374
name: Torch quantization nightly tests
@@ -485,56 +485,56 @@ jobs:
485485
name: torch_cuda_pipeline_level_quant_reports
486486
path: reports
487487

488-
run_flax_tpu_tests:
489-
name: Nightly Flax TPU Tests
490-
runs-on:
491-
group: gcp-ct5lp-hightpu-8t
492-
if: github.event_name == 'schedule'
493-
494-
container:
495-
image: diffusers/diffusers-flax-tpu
496-
options: --shm-size "16gb" --ipc host --privileged ${{ vars.V5_LITEPOD_8_ENV}} -v /mnt/hf_cache:/mnt/hf_cache
497-
defaults:
498-
run:
499-
shell: bash
500-
steps:
501-
- name: Checkout diffusers
502-
uses: actions/checkout@v3
503-
with:
504-
fetch-depth: 2
505-
506-
- name: Install dependencies
507-
run: |
508-
python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH"
509-
python -m uv pip install -e [quality,test]
510-
pip uninstall accelerate -y && python -m uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git
511-
python -m uv pip install pytest-reportlog
512-
513-
- name: Environment
514-
run: python utils/print_env.py
515-
516-
- name: Run nightly Flax TPU tests
517-
env:
518-
HF_TOKEN: ${{ secrets.DIFFUSERS_HF_HUB_READ_TOKEN }}
519-
run: |
520-
python -m pytest -n 0 \
521-
-s -v -k "Flax" \
522-
--make-reports=tests_flax_tpu \
523-
--report-log=tests_flax_tpu.log \
524-
tests/
525-
526-
- name: Failure short reports
527-
if: ${{ failure() }}
528-
run: |
529-
cat reports/tests_flax_tpu_stats.txt
530-
cat reports/tests_flax_tpu_failures_short.txt
531-
532-
- name: Test suite reports artifacts
533-
if: ${{ always() }}
534-
uses: actions/upload-artifact@v4
535-
with:
536-
name: flax_tpu_test_reports
537-
path: reports
488+
# run_flax_tpu_tests:
489+
# name: Nightly Flax TPU Tests
490+
# runs-on:
491+
# group: gcp-ct5lp-hightpu-8t
492+
# if: github.event_name == 'schedule'
493+
494+
# container:
495+
# image: diffusers/diffusers-flax-tpu
496+
# options: --shm-size "16gb" --ipc host --privileged ${{ vars.V5_LITEPOD_8_ENV}} -v /mnt/hf_cache:/mnt/hf_cache
497+
# defaults:
498+
# run:
499+
# shell: bash
500+
# steps:
501+
# - name: Checkout diffusers
502+
# uses: actions/checkout@v3
503+
# with:
504+
# fetch-depth: 2
505+
506+
# - name: Install dependencies
507+
# run: |
508+
# python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH"
509+
# python -m uv pip install -e [quality,test]
510+
# pip uninstall accelerate -y && python -m uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git
511+
# python -m uv pip install pytest-reportlog
512+
513+
# - name: Environment
514+
# run: python utils/print_env.py
515+
516+
# - name: Run nightly Flax TPU tests
517+
# env:
518+
# HF_TOKEN: ${{ secrets.DIFFUSERS_HF_HUB_READ_TOKEN }}
519+
# run: |
520+
# python -m pytest -n 0 \
521+
# -s -v -k "Flax" \
522+
# --make-reports=tests_flax_tpu \
523+
# --report-log=tests_flax_tpu.log \
524+
# tests/
525+
526+
# - name: Failure short reports
527+
# if: ${{ failure() }}
528+
# run: |
529+
# cat reports/tests_flax_tpu_stats.txt
530+
# cat reports/tests_flax_tpu_failures_short.txt
531+
532+
# - name: Test suite reports artifacts
533+
# if: ${{ always() }}
534+
# uses: actions/upload-artifact@v4
535+
# with:
536+
# name: flax_tpu_test_reports
537+
# path: reports
538538

539539
generate_consolidated_report:
540540
name: Generate Consolidated Test Report
@@ -545,9 +545,9 @@ jobs:
545545
run_big_gpu_torch_tests,
546546
run_nightly_quantization_tests,
547547
run_nightly_pipeline_level_quantization_tests,
548-
run_nightly_onnx_tests,
548+
# run_nightly_onnx_tests,
549549
torch_minimum_version_cuda_tests,
550-
run_flax_tpu_tests
550+
# run_flax_tpu_tests
551551
]
552552
if: always()
553553
runs-on:

‎.github/workflows/pr_tests.yml‎

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -87,11 +87,11 @@ jobs:
8787
runner: aws-general-8-plus
8888
image: diffusers/diffusers-pytorch-cpu
8989
report: torch_cpu_models_schedulers
90-
- name: Fast Flax CPU tests
91-
framework: flax
92-
runner: aws-general-8-plus
93-
image: diffusers/diffusers-flax-cpu
94-
report: flax_cpu
90+
# - name: Fast Flax CPU tests
91+
# framework: flax
92+
# runner: aws-general-8-plus
93+
# image: diffusers/diffusers-flax-cpu
94+
# report: flax_cpu
9595
- name: PyTorch Example CPU tests
9696
framework: pytorch_examples
9797
runner: aws-general-8-plus
@@ -147,14 +147,14 @@ jobs:
147147
--make-reports=tests_${{ matrix.config.report }} \
148148
tests/models tests/schedulers tests/others
149149
150-
- name: Run fast Flax TPU tests
151-
if: ${{ matrix.config.framework == 'flax' }}
152-
run: |
153-
python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH"
154-
python -m pytest -n 4 --max-worker-restart=0 --dist=loadfile \
155-
-s -v -k "Flax" \
156-
--make-reports=tests_${{ matrix.config.report }} \
157-
tests
150+
# - name: Run fast Flax TPU tests
151+
# if: ${{ matrix.config.framework == 'flax' }}
152+
# run: |
153+
# python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH"
154+
# python -m pytest -n 4 --max-worker-restart=0 --dist=loadfile \
155+
# -s -v -k "Flax" \
156+
# --make-reports=tests_${{ matrix.config.report }} \
157+
# tests
158158

159159
- name: Run example PyTorch CPU tests
160160
if: ${{ matrix.config.framework == 'pytorch_examples' }}

0 commit comments

Comments
 (0)