Skip to content

Commit 2749d4b

Browse files
sayakpaulsayakpaul
committed
fix
2 parents 54a299c + 8c938fb commit 2749d4b

File tree

74 files changed

+3436
-689
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

74 files changed

+3436
-689
lines changed

‎.github/workflows/build_docker_images.yml‎

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -75,10 +75,6 @@ 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
8278
- diffusers-doc-builder
8379

8480
steps:

‎.github/workflows/nightly_tests.yml‎

Lines changed: 2 additions & 102 deletions
Original file line numberDiff line numberDiff line change
@@ -321,55 +321,6 @@ 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
372-
373324
run_nightly_quantization_tests:
374325
name: Torch quantization nightly tests
375326
strategy:
@@ -485,57 +436,6 @@ jobs:
485436
name: torch_cuda_pipeline_level_quant_reports
486437
path: reports
487438

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
538-
539439
generate_consolidated_report:
540440
name: Generate Consolidated Test Report
541441
needs: [
@@ -545,9 +445,9 @@ jobs:
545445
run_big_gpu_torch_tests,
546446
run_nightly_quantization_tests,
547447
run_nightly_pipeline_level_quantization_tests,
548-
run_nightly_onnx_tests,
448+
# run_nightly_onnx_tests,
549449
torch_minimum_version_cuda_tests,
550-
run_flax_tpu_tests
450+
# run_flax_tpu_tests
551451
]
552452
if: always()
553453
runs-on:

‎.github/workflows/pr_tests.yml‎

Lines changed: 0 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -87,11 +87,6 @@ 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
9590
- name: PyTorch Example CPU tests
9691
framework: pytorch_examples
9792
runner: aws-general-8-plus
@@ -147,15 +142,6 @@ jobs:
147142
--make-reports=tests_${{ matrix.config.report }} \
148143
tests/models tests/schedulers tests/others
149144
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
158-
159145
- name: Run example PyTorch CPU tests
160146
if: ${{ matrix.config.framework == 'pytorch_examples' }}
161147
run: |

‎.github/workflows/push_tests.yml‎

Lines changed: 0 additions & 96 deletions
Original file line numberDiff line numberDiff line change
@@ -159,102 +159,6 @@ jobs:
159159
name: torch_cuda_test_reports_${{ matrix.module }}
160160
path: reports
161161

162-
flax_tpu_tests:
163-
name: Flax TPU Tests
164-
runs-on:
165-
group: gcp-ct5lp-hightpu-8t
166-
container:
167-
image: diffusers/diffusers-flax-tpu
168-
options: --shm-size "16gb" --ipc host --privileged ${{ vars.V5_LITEPOD_8_ENV}} -v /mnt/hf_cache:/mnt/hf_cache
169-
defaults:
170-
run:
171-
shell: bash
172-
steps:
173-
- name: Checkout diffusers
174-
uses: actions/checkout@v3
175-
with:
176-
fetch-depth: 2
177-
178-
- name: Install dependencies
179-
run: |
180-
python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH"
181-
python -m uv pip install -e [quality,test]
182-
pip uninstall accelerate -y && python -m uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git
183-
184-
- name: Environment
185-
run: |
186-
python utils/print_env.py
187-
188-
- name: Run Flax TPU tests
189-
env:
190-
HF_TOKEN: ${{ secrets.DIFFUSERS_HF_HUB_READ_TOKEN }}
191-
run: |
192-
python -m pytest -n 0 \
193-
-s -v -k "Flax" \
194-
--make-reports=tests_flax_tpu \
195-
tests/
196-
197-
- name: Failure short reports
198-
if: ${{ failure() }}
199-
run: |
200-
cat reports/tests_flax_tpu_stats.txt
201-
cat reports/tests_flax_tpu_failures_short.txt
202-
203-
- name: Test suite reports artifacts
204-
if: ${{ always() }}
205-
uses: actions/upload-artifact@v4
206-
with:
207-
name: flax_tpu_test_reports
208-
path: reports
209-
210-
onnx_cuda_tests:
211-
name: ONNX CUDA Tests
212-
runs-on:
213-
group: aws-g4dn-2xlarge
214-
container:
215-
image: diffusers/diffusers-onnxruntime-cuda
216-
options: --shm-size "16gb" --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/ --gpus 0
217-
defaults:
218-
run:
219-
shell: bash
220-
steps:
221-
- name: Checkout diffusers
222-
uses: actions/checkout@v3
223-
with:
224-
fetch-depth: 2
225-
226-
- name: Install dependencies
227-
run: |
228-
python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH"
229-
python -m uv pip install -e [quality,test]
230-
pip uninstall accelerate -y && python -m uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git
231-
232-
- name: Environment
233-
run: |
234-
python utils/print_env.py
235-
236-
- name: Run ONNXRuntime CUDA tests
237-
env:
238-
HF_TOKEN: ${{ secrets.DIFFUSERS_HF_HUB_READ_TOKEN }}
239-
run: |
240-
python -m pytest -n 1 --max-worker-restart=0 --dist=loadfile \
241-
-s -v -k "Onnx" \
242-
--make-reports=tests_onnx_cuda \
243-
tests/
244-
245-
- name: Failure short reports
246-
if: ${{ failure() }}
247-
run: |
248-
cat reports/tests_onnx_cuda_stats.txt
249-
cat reports/tests_onnx_cuda_failures_short.txt
250-
251-
- name: Test suite reports artifacts
252-
if: ${{ always() }}
253-
uses: actions/upload-artifact@v4
254-
with:
255-
name: onnx_cuda_test_reports
256-
path: reports
257-
258162
run_torch_compile_tests:
259163
name: PyTorch Compile CUDA tests
260164

‎.github/workflows/push_tests_fast.yml‎

Lines changed: 0 additions & 28 deletions
Original file line numberDiff line numberDiff line change
@@ -33,16 +33,6 @@ jobs:
3333
runner: aws-general-8-plus
3434
image: diffusers/diffusers-pytorch-cpu
3535
report: torch_cpu
36-
- name: Fast Flax CPU tests on Ubuntu
37-
framework: flax
38-
runner: aws-general-8-plus
39-
image: diffusers/diffusers-flax-cpu
40-
report: flax_cpu
41-
- name: Fast ONNXRuntime CPU tests on Ubuntu
42-
framework: onnxruntime
43-
runner: aws-general-8-plus
44-
image: diffusers/diffusers-onnxruntime-cpu
45-
report: onnx_cpu
4636
- name: PyTorch Example CPU tests on Ubuntu
4737
framework: pytorch_examples
4838
runner: aws-general-8-plus
@@ -87,24 +77,6 @@ jobs:
8777
--make-reports=tests_${{ matrix.config.report }} \
8878
tests/
8979
90-
- name: Run fast Flax TPU tests
91-
if: ${{ matrix.config.framework == 'flax' }}
92-
run: |
93-
python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH"
94-
python -m pytest -n 4 --max-worker-restart=0 --dist=loadfile \
95-
-s -v -k "Flax" \
96-
--make-reports=tests_${{ matrix.config.report }} \
97-
tests/
98-
99-
- name: Run fast ONNXRuntime CPU tests
100-
if: ${{ matrix.config.framework == 'onnxruntime' }}
101-
run: |
102-
python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH"
103-
python -m pytest -n 4 --max-worker-restart=0 --dist=loadfile \
104-
-s -v -k "Onnx" \
105-
--make-reports=tests_${{ matrix.config.report }} \
106-
tests/
107-
10880
- name: Run example PyTorch CPU tests
10981
if: ${{ matrix.config.framework == 'pytorch_examples' }}
11082
run: |

‎.github/workflows/push_tests_mps.yml‎

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,7 @@
11
name: Fast mps tests on main
22

33
on:
4-
push:
5-
branches:
6-
- main
7-
paths:
8-
- "src/diffusers/**.py"
9-
- "tests/**.py"
4+
workflow_dispatch:
105

116
env:
127
DIFFUSERS_IS_CI: yes

0 commit comments

Comments
 (0)