Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
119 commits
Select commit Hold shift + click to select a range
d54622c
[Modular] Allow custom blocks to be saved to `local_dir` (#12381)
DN6 Oct 31, 2025
051c8a1
Fix Stable Diffusion 3.x pooled prompt embedding with multiple images…
friedrich Oct 31, 2025
5e8415a
Fix custom code loading in Automodel (#12571)
DN6 Nov 2, 2025
cdbf0ad
[modular] better warn message (#12573)
yiyixuxu Nov 2, 2025
8f80dda
[tests] add tests for flux modular (t2i, i2i, kontext) (#12566)
sayakpaul Nov 2, 2025
de6173c
[modular]pass hub_kwargs to load_config (#12577)
yiyixuxu Nov 3, 2025
1ec28a2
ulysses enabling in native attention path (#12563)
sywangyi Nov 3, 2025
325a950
Kandinsky 5.0 Docs fixes (#12582)
leffff Nov 4, 2025
ac5a1e2
[docs] sort doc (#12586)
sayakpaul Nov 4, 2025
7d5790f
implement vae dp for AutoencoderKL and AutoencoderKLWan
TmacAaron Nov 4, 2025
dcfb18a
[LoRA] add support for more Qwen LoRAs (#12581)
linoytsaban Nov 4, 2025
6c61cd0
extract same code in vae dp func
TmacAaron Nov 5, 2025
4aeeeb9
optimize blend method in tiled vae
TmacAaron Nov 5, 2025
8cfad75
fix world_size 1 bug when init parallel tiling
TmacAaron Nov 5, 2025
2511515
fix bug in vae_kl_wan
TmacAaron Nov 5, 2025
6198f8a
[Modular] Allow ModularPipeline to load from revisions (#12592)
DN6 Nov 6, 2025
58f3771
Add optional precision-preserving preprocessing for examples/uncondit…
turian Nov 6, 2025
b3e9dfc
[SANA-Video] Adding 5s pre-trained 480p SANA-Video inference (#12584)
lawrence-cj Nov 6, 2025
e4393fa
Fix overflow and dtype handling in rgblike_to_depthmap (NumPy + PyTor…
MohammadSadeghSalehi Nov 6, 2025
8ac17cd
[Modular] Some clean up for Modular tests (#12579)
DN6 Nov 7, 2025
9f669e7
feat: enable attention dispatch for huanyuan video (#12591)
DefTruth Nov 7, 2025
a9cb08a
fix the crash in Wan-AI/Wan2.2-TI2V-5B-Diffusers if CP is enabled (#1…
sywangyi Nov 7, 2025
bc8fd86
[CI] Push test fix (#12617)
DN6 Nov 10, 2025
04f9d2b
add ChronoEdit (#12593)
zhangjiewu Nov 10, 2025
b455dc9
[modular] wan! (#12611)
yiyixuxu Nov 10, 2025
35d7033
[CI] Fix typo in uv install (#12618)
DN6 Nov 10, 2025
0fd58c7
fix: correct import path for load_model_dict_into_meta in conversion …
yashwantbezawada Nov 10, 2025
8d45f21
Fix Context Parallel validation checks (#12446)
DN6 Nov 10, 2025
8f6328c
[Modular] Clean up docs (#12604)
DN6 Nov 10, 2025
5a47442
Fix: update type hints for Tuple parameters across multiple files to …
cesaryuan Nov 10, 2025
66e6a02
[CI] Remove unittest dependency from `testing_utils.py` (#12621)
DN6 Nov 11, 2025
ff26394
Fix rotary positional embedding dimension mismatch in Wan and SkyReel…
charchit7 Nov 11, 2025
0c75892
fix copies (#12637)
yiyixuxu Nov 12, 2025
aecf0c5
Add MLU Support. (#12629)
a120092009 Nov 12, 2025
093cd3f
fix dispatch_attention_fn check (#12636)
yiyixuxu Nov 12, 2025
f5e5f34
[modular] add tests for qwen modular (#12585)
sayakpaul Nov 12, 2025
f3db38c
ArXiv -> HF Papers (#12583)
qgallouedec Nov 12, 2025
2f44d63
[docs] Update install instructions (#12626)
stevhliu Nov 12, 2025
d6c63bb
[modular] add a check (#12628)
yiyixuxu Nov 12, 2025
44c3101
Improve docstrings and type hints in scheduling_amused.py (#12623)
delmalih Nov 13, 2025
d8e4805
[WIP]Add Wan2.2 Animate Pipeline (Continuation of #12442 by tolgacang…
dg845 Nov 13, 2025
7a001c3
adjust unit tests for `test_save_load_float16` (#12500)
kaixuanliu Nov 13, 2025
cd3bbe2
skip autoencoderdl layerwise casting memory (#12647)
sayakpaul Nov 13, 2025
6a2309b
[utils] Update check_doc_toc (#12642)
stevhliu Nov 13, 2025
40de88a
[docs] AutoModel (#12644)
stevhliu Nov 13, 2025
6fe4a6f
Improve docstrings and type hints in scheduling_ddim.py (#12622)
delmalih Nov 13, 2025
3c1ca86
Improve docstrings and type hints in scheduling_ddpm.py (#12651)
delmalih Nov 13, 2025
eeae033
[Modular] Add Custom Blocks guide to doc (#12339)
DN6 Nov 14, 2025
63dd601
Improve docstrings and type hints in scheduling_euler_discrete.py (#1…
delmalih Nov 14, 2025
a9e4883
Update Wan Animate Docs (#12658)
dg845 Nov 15, 2025
01a5692
Rope in float32 for mps or npu compatibility (#12665)
DavidBert Nov 15, 2025
0c35b58
[PRX pipeline]: add 1024 resolution ratio bins (#12670)
DavidBert Nov 17, 2025
1afc218
SANA-Video Image to Video pipeline `SanaImageToVideoPipeline` support…
lawrence-cj Nov 17, 2025
3579fda
[CI] Make CI logs less verbose (#12674)
DN6 Nov 17, 2025
67dc65e
Revert `AutoencoderKLWan`'s `dim_mult` default value back to list (#1…
dg845 Nov 17, 2025
b7df4a5
[CI] Temporarily pin transformers (#12677)
DN6 Nov 18, 2025
ab71f3c
[core] Refactor hub attn kernels (#12475)
sayakpaul Nov 19, 2025
6d8973f
[CI] Fix indentation issue in workflow files (#12685)
DN6 Nov 19, 2025
a96b145
[CI] Fix failing Pipeline CPU tests (#12681)
DN6 Nov 19, 2025
15370f8
Improve docstrings and type hints in scheduling_pndm.py (#12676)
delmalih Nov 19, 2025
d5da453
Community Pipeline: FluxFillControlNetInpaintPipeline for FLUX Fill-B…
pratim4dasude Nov 19, 2025
6f1042e
Improve docstrings and type hints in scheduling_lms_discrete.py (#12678)
delmalih Nov 21, 2025
544ba67
Add FluxLoraLoaderMixin to Fibo pipeline (#12688)
SwayStar123 Nov 24, 2025
354d35a
bugfix: fix chrono-edit context parallel (#12660)
DefTruth Nov 24, 2025
d176f61
[core] support sage attention + FA2 through `kernels` (#12439)
sayakpaul Nov 24, 2025
fbcd3ba
[i8n-pt] Fix grammar and expand Portuguese documentation (#12598)
cdutr Nov 24, 2025
dde8754
Fix variable naming typos in community FluxControlNetFillInpaintPipel…
sqhuang Nov 24, 2025
d33d9f6
fix typo in docs (#12675)
lawrence-cj Nov 25, 2025
4088e8a
Add Support for Z-Image Series (#12703)
JerryWu-code Nov 25, 2025
5ffb73d
let's go Flux2 🚀 (#12711)
sayakpaul Nov 25, 2025
ac78646
Update script names in README for Flux2 training (#12713)
anvilarth Nov 26, 2025
b91e8c0
[lora]: Fix Flux2 LoRA NaN test (#12714)
sayakpaul Nov 26, 2025
94c9613
[docs] Correct flux2 links (#12716)
sayakpaul Nov 26, 2025
c8656ed
[docs] put autopipeline after overview and hunyuanimage in images (#1…
sayakpaul Nov 26, 2025
a88a7b4
Improve docstrings and type hints in scheduling_dpmsolver_multistep.p…
delmalih Nov 26, 2025
e6d4612
Support unittest for Z-image ⚡️ (#12715)
JerryWu-code Nov 26, 2025
6bf668c
[chore] remove torch.save from remnant code. (#12717)
sayakpaul Nov 27, 2025
01e3555
Enable regional compilation on z-image transformer model (#12736)
sayakpaul Nov 27, 2025
1b91856
Fix examples not loading LoRA adapter weights from checkpoint (#12690)
SurAyush Nov 28, 2025
b010a8c
[Modular] Add single file support to Modular (#12383)
DN6 Nov 28, 2025
152f7ca
fix type-check for z-image transformer (#12739)
DefTruth Nov 29, 2025
6156cf8
Hunyuanvideo15 (#12696)
yiyixuxu Dec 1, 2025
c25582d
[Docs] Update Imagen Video paper link in schedulers (#12724)
delmalih Dec 1, 2025
d769d8a
Improve docstrings and type hints in scheduling_heun_discrete.py (#12…
delmalih Dec 1, 2025
859b809
Improve docstrings and type hints in scheduling_euler_ancestral_discr…
delmalih Dec 1, 2025
fffd964
fix FLUX.2 context parallel (#12737)
DefTruth Dec 1, 2025
99784ae
Rename BriaPipeline to BriaFiboPipeline in documentation (#12758)
galbria Dec 1, 2025
394a48d
Update bria_fibo.md with minor fixes (#12731)
sayakpaul Dec 1, 2025
564079f
[feat]: implement "local" caption upsampling for Flux.2 (#12718)
sayakpaul Dec 1, 2025
edf36f5
Add ZImage LoRA support and integrate into ZImagePipeline (#12750)
CalamitousFelicitousness Dec 2, 2025
4f136f8
Add support for Ovis-Image (#12740)
DoctorKey Dec 2, 2025
9379b23
Fix TPU (torch_xla) compatibility Error about tensor repeat func alon…
JerryWu-code Dec 2, 2025
3c05b9f
Fixes #12673. `record_stream` in group offloading is not working prop…
KimbingNg Dec 3, 2025
f48f9c2
[core] start varlen variants for attn backend kernels. (#12765)
sayakpaul Dec 3, 2025
759ea58
[core] reuse `AttentionMixin` for compatible classes (#12463)
sayakpaul Dec 3, 2025
1908c47
Deprecate `upcast_vae` in SDXL based pipelines (#12619)
DN6 Dec 3, 2025
d0c54e5
Kandinsky 5.0 Video Pro and Image Lite (#12664)
leffff Dec 3, 2025
5ab5946
Fix: leaf_level offloading breaks after delete_adapters (#12639)
adi776borate Dec 3, 2025
d96cbac
[tests] fix hunuyanvideo 1.5 offloading tests. (#12782)
sayakpaul Dec 3, 2025
a1f36ee
[Z-Image] various small changes, Z-Image transformer tests, etc. (#12…
sayakpaul Dec 3, 2025
6028613
Z-Image-Turbo `from_single_file` (#12756)
hlky Dec 4, 2025
c318686
Update attention_backends.md to format kernels (#12757)
sayakpaul Dec 4, 2025
2842c14
Improve docstrings and type hints in scheduling_unipc_multistep.py (#…
delmalih Dec 4, 2025
cd00ba6
fix spatial compression ratio error for AutoEncoderKLWan doing tiled …
jerry2102 Dec 4, 2025
7de51b8
[lora] support more ZImage LoRAs (#12790)
sayakpaul Dec 4, 2025
8d415a6
PRX Set downscale_freq_shift to 0 for consistency with internal imple…
DavidBert Dec 4, 2025
f12d161
Fix broken group offloading with block_level for models with standalo…
rycerzes Dec 5, 2025
c98c157
[Docs] Add Z-Image docs (#12775)
asomoza Dec 5, 2025
bb9e713
move kandisnky docs.
sayakpaul Dec 5, 2025
8430ac2
[docs] minor fixes to kandinsky docs (#12797)
sayakpaul Dec 5, 2025
256e010
Improve docstrings and type hints in scheduling_deis_multistep.py (#1…
delmalih Dec 5, 2025
6290fdf
[Feat] TaylorSeer Cache (#12648)
toilaluan Dec 6, 2025
5a74319
Update the TensorRT-ModelOPT to Nvidia-ModelOPT (#12793)
jingyu-ml Dec 8, 2025
f67639b
add post init for safty checker (#12794)
jiqing-feng Dec 8, 2025
671149e
[HunyuanVideo1.5] support step-distilled (#12802)
yiyixuxu Dec 8, 2025
2246d2c
Add ZImageImg2ImgPipeline (#12751)
CalamitousFelicitousness Dec 8, 2025
3d02cd5
[PRX] Improve model compilation (#12787)
WaterKnight1998 Dec 8, 2025
54fa074
Improve docstrings and type hints in scheduling_dpmsolver_singlestep.…
delmalih Dec 8, 2025
43b1722
Merge branch 'ascend-patch' into vae_dp
TmacAaron Dec 9, 2025
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
The table of contents is too big for display.
Diff view
Diff view
  •  
  •  
  •  
28 changes: 21 additions & 7 deletions .github/workflows/nightly_tests.yml
Original file line number Diff line number Diff line change
Expand Up @@ -73,6 +73,8 @@ jobs:
run: |
uv pip install -e ".[quality]"
uv pip uninstall accelerate && uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git
#uv pip uninstall transformers huggingface_hub && uv pip install --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git
uv pip uninstall transformers huggingface_hub && uv pip install transformers==4.57.1
uv pip install pytest-reportlog
- name: Environment
run: |
Expand All @@ -84,7 +86,7 @@ jobs:
CUBLAS_WORKSPACE_CONFIG: :16:8
run: |
pytest -n 1 --max-worker-restart=0 --dist=loadfile \
-s -v -k "not Flax and not Onnx" \
-k "not Flax and not Onnx" \
--make-reports=tests_pipeline_${{ matrix.module }}_cuda \
--report-log=tests_pipeline_${{ matrix.module }}_cuda.log \
tests/pipelines/${{ matrix.module }}
Expand Down Expand Up @@ -126,6 +128,8 @@ jobs:
uv pip install -e ".[quality]"
uv pip install peft@git+https://github.com/huggingface/peft.git
uv pip uninstall accelerate && uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git
#uv pip uninstall transformers huggingface_hub && uv pip install --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git
uv pip uninstall transformers huggingface_hub && uv pip install transformers==4.57.1
uv pip install pytest-reportlog
- name: Environment
run: python utils/print_env.py
Expand All @@ -138,7 +142,7 @@ jobs:
CUBLAS_WORKSPACE_CONFIG: :16:8
run: |
pytest -n 1 --max-worker-restart=0 --dist=loadfile \
-s -v -k "not Flax and not Onnx" \
-k "not Flax and not Onnx" \
--make-reports=tests_torch_${{ matrix.module }}_cuda \
--report-log=tests_torch_${{ matrix.module }}_cuda.log \
tests/${{ matrix.module }}
Expand All @@ -151,7 +155,7 @@ jobs:
CUBLAS_WORKSPACE_CONFIG: :16:8
run: |
pytest -n 1 --max-worker-restart=0 --dist=loadfile \
-s -v --make-reports=examples_torch_cuda \
--make-reports=examples_torch_cuda \
--report-log=examples_torch_cuda.log \
examples/

Expand Down Expand Up @@ -190,6 +194,8 @@ jobs:
- name: Install dependencies
run: |
uv pip install -e ".[quality,training]"
#uv pip uninstall transformers huggingface_hub && uv pip install --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git
uv pip uninstall transformers huggingface_hub && uv pip install transformers==4.57.1
- name: Environment
run: |
python utils/print_env.py
Expand All @@ -198,7 +204,7 @@ jobs:
HF_TOKEN: ${{ secrets.DIFFUSERS_HF_HUB_READ_TOKEN }}
RUN_COMPILE: yes
run: |
pytest -n 1 --max-worker-restart=0 --dist=loadfile -s -v -k "compile" --make-reports=tests_torch_compile_cuda tests/
pytest -n 1 --max-worker-restart=0 --dist=loadfile -k "compile" --make-reports=tests_torch_compile_cuda tests/
- name: Failure short reports
if: ${{ failure() }}
run: cat reports/tests_torch_compile_cuda_failures_short.txt
Expand Down Expand Up @@ -232,6 +238,8 @@ jobs:
uv pip install -e ".[quality]"
uv pip install peft@git+https://github.com/huggingface/peft.git
uv pip uninstall accelerate && uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git
#uv pip uninstall transformers huggingface_hub && uv pip install --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git
uv pip uninstall transformers huggingface_hub && uv pip install transformers==4.57.1
uv pip install pytest-reportlog
- name: Environment
run: |
Expand Down Expand Up @@ -281,6 +289,8 @@ jobs:
uv pip install -e ".[quality]"
uv pip install peft@git+https://github.com/huggingface/peft.git
uv pip uninstall accelerate && uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git
#uv pip uninstall transformers huggingface_hub && uv pip install --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git
uv pip uninstall transformers huggingface_hub && uv pip install transformers==4.57.1

- name: Environment
run: |
Expand All @@ -293,7 +303,7 @@ jobs:
CUBLAS_WORKSPACE_CONFIG: :16:8
run: |
pytest -n 1 --max-worker-restart=0 --dist=loadfile \
-s -v -k "not Flax and not Onnx" \
-k "not Flax and not Onnx" \
--make-reports=tests_torch_minimum_version_cuda \
tests/models/test_modeling_common.py \
tests/pipelines/test_pipelines_common.py \
Expand Down Expand Up @@ -358,6 +368,8 @@ jobs:
uv pip install ${{ join(matrix.config.additional_deps, ' ') }}
fi
uv pip install pytest-reportlog
#uv pip uninstall transformers huggingface_hub && uv pip install --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git
uv pip uninstall transformers huggingface_hub && uv pip install transformers==4.57.1
- name: Environment
run: |
python utils/print_env.py
Expand Down Expand Up @@ -405,6 +417,8 @@ jobs:
run: |
uv pip install -e ".[quality]"
uv pip install -U bitsandbytes optimum_quanto
#uv pip uninstall transformers huggingface_hub && uv pip install --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git
uv pip uninstall transformers huggingface_hub && uv pip install transformers==4.57.1
uv pip install pytest-reportlog
- name: Environment
run: |
Expand Down Expand Up @@ -531,7 +545,7 @@ jobs:
# HF_HOME: /System/Volumes/Data/mnt/cache
# HF_TOKEN: ${{ secrets.DIFFUSERS_HF_HUB_READ_TOKEN }}
# run: |
# ${CONDA_RUN} pytest -n 1 -s -v --make-reports=tests_torch_mps \
# ${CONDA_RUN} pytest -n 1 --make-reports=tests_torch_mps \
# --report-log=tests_torch_mps.log \
# tests/
# - name: Failure short reports
Expand Down Expand Up @@ -587,7 +601,7 @@ jobs:
# HF_HOME: /System/Volumes/Data/mnt/cache
# HF_TOKEN: ${{ secrets.DIFFUSERS_HF_HUB_READ_TOKEN }}
# run: |
# ${CONDA_RUN} pytest -n 1 -s -v --make-reports=tests_torch_mps \
# ${CONDA_RUN} pytest -n 1 --make-reports=tests_torch_mps \
# --report-log=tests_torch_mps.log \
# tests/
# - name: Failure short reports
Expand Down
5 changes: 3 additions & 2 deletions .github/workflows/pr_modular_tests.yml
Original file line number Diff line number Diff line change
Expand Up @@ -109,7 +109,8 @@ jobs:
- name: Install dependencies
run: |
uv pip install -e ".[quality]"
uv pip uninstall transformers huggingface_hub && uv pip install --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git
#uv pip uninstall transformers huggingface_hub && uv pip install --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git
uv pip uninstall transformers huggingface_hub && uv pip install transformers==4.57.1
uv pip uninstall accelerate && uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git --no-deps

- name: Environment
Expand All @@ -120,7 +121,7 @@ jobs:
if: ${{ matrix.config.framework == 'pytorch_pipelines' }}
run: |
pytest -n 8 --max-worker-restart=0 --dist=loadfile \
-s -v -k "not Flax and not Onnx" \
-k "not Flax and not Onnx" \
--make-reports=tests_${{ matrix.config.report }} \
tests/modular_pipelines

Expand Down
14 changes: 8 additions & 6 deletions .github/workflows/pr_tests.yml
Original file line number Diff line number Diff line change
Expand Up @@ -115,7 +115,8 @@ jobs:
- name: Install dependencies
run: |
uv pip install -e ".[quality]"
uv pip uninstall transformers huggingface_hub && uv pip install --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git
#uv pip uninstall transformers huggingface_hub && uv pip install --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git
uv pip uninstall transformers huggingface_hub && uv pip install transformers==4.57.1
uv pip uninstall accelerate && uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git --no-deps

- name: Environment
Expand All @@ -126,15 +127,15 @@ jobs:
if: ${{ matrix.config.framework == 'pytorch_pipelines' }}
run: |
pytest -n 8 --max-worker-restart=0 --dist=loadfile \
-s -v -k "not Flax and not Onnx" \
-k "not Flax and not Onnx" \
--make-reports=tests_${{ matrix.config.report }} \
tests/pipelines

- name: Run fast PyTorch Model Scheduler CPU tests
if: ${{ matrix.config.framework == 'pytorch_models' }}
run: |
pytest -n 4 --max-worker-restart=0 --dist=loadfile \
-s -v -k "not Flax and not Onnx and not Dependency" \
-k "not Flax and not Onnx and not Dependency" \
--make-reports=tests_${{ matrix.config.report }} \
tests/models tests/schedulers tests/others

Expand Down Expand Up @@ -246,7 +247,8 @@ jobs:
uv pip install -U peft@git+https://github.com/huggingface/peft.git --no-deps
uv pip install -U tokenizers
uv pip uninstall accelerate && uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git --no-deps
uv pip uninstall transformers huggingface_hub && uv pip install --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git
#uv pip uninstall transformers huggingface_hub && uv pip install --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git
uv pip uninstall transformers huggingface_hub && uv pip install transformers==4.57.1

- name: Environment
run: |
Expand All @@ -255,11 +257,11 @@ jobs:
- name: Run fast PyTorch LoRA tests with PEFT
run: |
pytest -n 4 --max-worker-restart=0 --dist=loadfile \
-s -v \
\
--make-reports=tests_peft_main \
tests/lora/
pytest -n 4 --max-worker-restart=0 --dist=loadfile \
-s -v \
\
--make-reports=tests_models_lora_peft_main \
tests/models/ -k "lora"

Expand Down
33 changes: 18 additions & 15 deletions .github/workflows/pr_tests_gpu.yml
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
name: Fast GPU Tests on PR
name: Fast GPU Tests on PR

on:
pull_request:
Expand Down Expand Up @@ -71,7 +71,7 @@ jobs:
if: ${{ failure() }}
run: |
echo "Repo consistency check failed. Please ensure the right dependency versions are installed with 'pip install -e .[quality]' and run 'make fix-copies'" >> $GITHUB_STEP_SUMMARY

setup_torch_cuda_pipeline_matrix:
needs: [check_code_quality, check_repository_consistency]
name: Setup Torch Pipelines CUDA Slow Tests Matrix
Expand Down Expand Up @@ -131,7 +131,8 @@ jobs:
run: |
uv pip install -e ".[quality]"
uv pip uninstall accelerate && uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git
uv pip uninstall transformers huggingface_hub && uv pip install --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git
#uv pip uninstall transformers huggingface_hub && uv pip install --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git
uv pip uninstall transformers huggingface_hub && uv pip install transformers==4.57.1

- name: Environment
run: |
Expand All @@ -149,18 +150,18 @@ jobs:
# https://pytorch.org/docs/stable/notes/randomness.html#avoiding-nondeterministic-algorithms
CUBLAS_WORKSPACE_CONFIG: :16:8
run: |
if [ "${{ matrix.module }}" = "ip_adapters" ]; then
if [ "${{ matrix.module }}" = "ip_adapters" ]; then
pytest -n 1 --max-worker-restart=0 --dist=loadfile \
-s -v -k "not Flax and not Onnx" \
-k "not Flax and not Onnx" \
--make-reports=tests_pipeline_${{ matrix.module }}_cuda \
tests/pipelines/${{ matrix.module }}
else
else
pattern=$(cat ${{ steps.extract_tests.outputs.pattern_file }})
pytest -n 1 --max-worker-restart=0 --dist=loadfile \
-s -v -k "not Flax and not Onnx and $pattern" \
-k "not Flax and not Onnx and $pattern" \
--make-reports=tests_pipeline_${{ matrix.module }}_cuda \
tests/pipelines/${{ matrix.module }}
fi
fi

- name: Failure short reports
if: ${{ failure() }}
Expand Down Expand Up @@ -201,7 +202,8 @@ jobs:
uv pip install -e ".[quality]"
uv pip install peft@git+https://github.com/huggingface/peft.git
uv pip uninstall accelerate && uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git
uv pip uninstall transformers huggingface_hub && uv pip install --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git
#uv pip uninstall transformers huggingface_hub && uv pip install --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git
uv pip uninstall transformers huggingface_hub && uv pip install transformers==4.57.1

- name: Environment
run: |
Expand All @@ -222,11 +224,11 @@ jobs:
run: |
pattern=$(cat ${{ steps.extract_tests.outputs.pattern_file }})
if [ -z "$pattern" ]; then
pytest -n 1 -sv --max-worker-restart=0 --dist=loadfile -k "not Flax and not Onnx" tests/${{ matrix.module }} \
--make-reports=tests_torch_cuda_${{ matrix.module }}
pytest -n 1 --max-worker-restart=0 --dist=loadfile -k "not Flax and not Onnx" tests/${{ matrix.module }} \
--make-reports=tests_torch_cuda_${{ matrix.module }}
else
pytest -n 1 -sv --max-worker-restart=0 --dist=loadfile -k "not Flax and not Onnx and $pattern" tests/${{ matrix.module }} \
--make-reports=tests_torch_cuda_${{ matrix.module }}
pytest -n 1 --max-worker-restart=0 --dist=loadfile -k "not Flax and not Onnx and $pattern" tests/${{ matrix.module }} \
--make-reports=tests_torch_cuda_${{ matrix.module }}
fi

- name: Failure short reports
Expand Down Expand Up @@ -262,7 +264,8 @@ jobs:
nvidia-smi
- name: Install dependencies
run: |
uv pip uninstall transformers huggingface_hub && uv pip install --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git
#uv pip uninstall transformers huggingface_hub && uv pip install --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git
uv pip uninstall transformers huggingface_hub && uv pip install transformers==4.57.1
uv pip install -e ".[quality,training]"

- name: Environment
Expand All @@ -274,7 +277,7 @@ jobs:
HF_TOKEN: ${{ secrets.DIFFUSERS_HF_HUB_READ_TOKEN }}
run: |
uv pip install ".[training]"
pytest -n 1 --max-worker-restart=0 --dist=loadfile -s -v --make-reports=examples_torch_cuda examples/
pytest -n 1 --max-worker-restart=0 --dist=loadfile --make-reports=examples_torch_cuda examples/

- name: Failure short reports
if: ${{ failure() }}
Expand Down
16 changes: 11 additions & 5 deletions .github/workflows/push_tests.yml
Original file line number Diff line number Diff line change
Expand Up @@ -76,6 +76,8 @@ jobs:
run: |
uv pip install -e ".[quality]"
uv pip uninstall accelerate && uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git
#uv pip uninstall transformers huggingface_hub && uv pip install --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git
uv pip uninstall transformers huggingface_hub && uv pip install transformers==4.57.1
- name: Environment
run: |
python utils/print_env.py
Expand All @@ -86,7 +88,7 @@ jobs:
CUBLAS_WORKSPACE_CONFIG: :16:8
run: |
pytest -n 1 --max-worker-restart=0 --dist=loadfile \
-s -v -k "not Flax and not Onnx" \
-k "not Flax and not Onnx" \
--make-reports=tests_pipeline_${{ matrix.module }}_cuda \
tests/pipelines/${{ matrix.module }}
- name: Failure short reports
Expand Down Expand Up @@ -127,6 +129,8 @@ jobs:
uv pip install -e ".[quality]"
uv pip install peft@git+https://github.com/huggingface/peft.git
uv pip uninstall accelerate && uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git
#uv pip uninstall transformers huggingface_hub && uv pip install --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git
uv pip uninstall transformers huggingface_hub && uv pip install transformers==4.57.1

- name: Environment
run: |
Expand All @@ -139,7 +143,7 @@ jobs:
CUBLAS_WORKSPACE_CONFIG: :16:8
run: |
pytest -n 1 --max-worker-restart=0 --dist=loadfile \
-s -v -k "not Flax and not Onnx" \
-k "not Flax and not Onnx" \
--make-reports=tests_torch_cuda_${{ matrix.module }} \
tests/${{ matrix.module }}

Expand Down Expand Up @@ -178,6 +182,8 @@ jobs:
- name: Install dependencies
run: |
uv pip install -e ".[quality,training]"
#uv pip uninstall transformers huggingface_hub && uv pip install --prerelease allow -U transformers@git+https://github.com/huggingface/transformers.git
uv pip uninstall transformers huggingface_hub && uv pip install transformers==4.57.1
- name: Environment
run: |
python utils/print_env.py
Expand All @@ -186,7 +192,7 @@ jobs:
HF_TOKEN: ${{ secrets.DIFFUSERS_HF_HUB_READ_TOKEN }}
RUN_COMPILE: yes
run: |
pytest -n 1 --max-worker-restart=0 --dist=loadfile -s -v -k "compile" --make-reports=tests_torch_compile_cuda tests/
pytest -n 1 --max-worker-restart=0 --dist=loadfile -k "compile" --make-reports=tests_torch_compile_cuda tests/
- name: Failure short reports
if: ${{ failure() }}
run: cat reports/tests_torch_compile_cuda_failures_short.txt
Expand Down Expand Up @@ -227,7 +233,7 @@ jobs:
env:
HF_TOKEN: ${{ secrets.DIFFUSERS_HF_HUB_READ_TOKEN }}
run: |
pytest -n 1 --max-worker-restart=0 --dist=loadfile -s -v -k "xformers" --make-reports=tests_torch_xformers_cuda tests/
pytest -n 1 --max-worker-restart=0 --dist=loadfile -k "xformers" --make-reports=tests_torch_xformers_cuda tests/
- name: Failure short reports
if: ${{ failure() }}
run: cat reports/tests_torch_xformers_cuda_failures_short.txt
Expand Down Expand Up @@ -270,7 +276,7 @@ jobs:
HF_TOKEN: ${{ secrets.DIFFUSERS_HF_HUB_READ_TOKEN }}
run: |
uv pip install ".[training]"
pytest -n 1 --max-worker-restart=0 --dist=loadfile -s -v --make-reports=examples_torch_cuda examples/
pytest -n 1 --max-worker-restart=0 --dist=loadfile --make-reports=examples_torch_cuda examples/

- name: Failure short reports
if: ${{ failure() }}
Expand Down
2 changes: 1 addition & 1 deletion .github/workflows/push_tests_fast.yml
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,7 @@ jobs:
if: ${{ matrix.config.framework == 'pytorch' }}
run: |
pytest -n 4 --max-worker-restart=0 --dist=loadfile \
-s -v -k "not Flax and not Onnx" \
-k "not Flax and not Onnx" \
--make-reports=tests_${{ matrix.config.report }} \
tests/

Expand Down
2 changes: 1 addition & 1 deletion .github/workflows/push_tests_mps.yml
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,7 @@ jobs:
HF_HOME: /System/Volumes/Data/mnt/cache
HF_TOKEN: ${{ secrets.HF_TOKEN }}
run: |
${CONDA_RUN} python -m pytest -n 0 -s -v --make-reports=tests_torch_mps tests/
${CONDA_RUN} python -m pytest -n 0 --make-reports=tests_torch_mps tests/

- name: Failure short reports
if: ${{ failure() }}
Expand Down
Loading