Commit Graph

4540 Commits

Author SHA1 Message Date
Leo Jiang 64dd1c2f2a Fix the issue on sd3 dreambooth w./w.t. lora training (#9419)
* Fix dtype error

* [bugfix] Fixed the issue on sd3 dreambooth training

* [bugfix] Fixed the issue on sd3 dreambooth training

---------

Co-authored-by: 蒋硕 <jiangshuo9@h-partners.com>
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
2024-12-23 13:02:16 +05:30
Benjamin Bossan a76767b684 MAINT Permission for GH token in stale.yml (#9427)
* MAINT Permission for GH token in stale.yml

See https://github.com/huggingface/peft/pull/2061 for the equivalent PR
in PEFT.

This restores the functionality of the stale bot after permissions for
the token have been limited. The action still shows errors for PEFT but
the bot appears to work fine.

* Also add write permissions for PRs

---------

Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
2024-12-23 13:02:16 +05:30
Sayak Paul 457ff5dbf8 [LoRA] fix adapter movement when using DoRA. (#9411)
fix adapter movement when using DoRA.
2024-12-23 13:02:16 +05:30
Dhruv Nair 4bc1d700cc [CI] Nightly Test Updates (#9380)
* update

* update

* update

* update

* update

---------

Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
Co-authored-by: YiYi Xu <yixu310@gmail.com>
2024-12-23 13:02:16 +05:30
Sayak Paul ca1692c2c4 [CI] update artifact uploader version (#9426)
update artifact uploader version
2024-12-23 13:02:16 +05:30
Juan Acevedo 04abe73b30 Ptxla sd training (#9381)
* enable pxla training of stable diffusion 2.x models.

* run linter/style and run pipeline test for stable diffusion and fix issues.

* update xla libraries

* fix read me newline.

* move files to research folder.

* update per comments.

* rename readme.

---------

Co-authored-by: Juan Acevedo <jfacevedo@google.com>
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
2024-12-23 13:02:16 +05:30
Aryan 5e6901d361 [docs] AnimateDiff FreeNoise (#9414)
* update docs

* apply suggestions from review

* Update docs/source/en/api/pipelines/animatediff.md

Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>

* Update docs/source/en/api/pipelines/animatediff.md

Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>

* Update docs/source/en/api/pipelines/animatediff.md

Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>

* apply suggestions from review

---------

Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>
2024-12-23 13:02:16 +05:30
asfiyab-nvidia c35b0f7dc4 FluxPosEmbed: Remove Squeeze No-op (#9409)
Remove Squeeze op

Signed-off-by: Asfiya Baig <asfiyab@nvidia.com>
Co-authored-by: YiYi Xu <yixu310@gmail.com>
2024-12-23 13:02:16 +05:30
dianyo e77c063247 Migrate the BrownianTree to BrownianInterval in DPM solver (#9335)
migrate the BrownianTree to BrownianInterval

Co-authored-by: YiYi Xu <yixu310@gmail.com>
2024-12-23 13:02:16 +05:30
Yu Zheng e587cb9866 [examples] add controlnet sd3 example (#9249)
* add controlnet sd3 example

* add controlnet sd3 example

* update controlnet sd3 example

* add controlnet sd3 example test

* fix quality and style

* update test

* update test

---------

Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
2024-12-23 13:02:16 +05:30
Sayak Paul 6f39b2bf0d [Tests] fix some fast gpu tests. (#9379)
fix some fast gpu tests.
2024-12-23 13:02:16 +05:30
captainzz faa57c4ff5 fix from_transformer() with extra conditioning channels (#9364)
* fix from_transformer() with extra conditioning channels

* style fix

---------

Co-authored-by: YiYi Xu <yixu310@gmail.com>
Co-authored-by: Álvaro Somoza <somoza.alvaro@gmail.com>
2024-12-23 13:02:16 +05:30
Jinzhe Pan 9f011c088b [docs] Add xDiT in section optimization (#9365)
* docs: add xDiT to optimization methods

* fix: picture layout problem

* docs: add more introduction about xdit & apply suggestions

* Apply suggestions from code review

Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>

---------

Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>
2024-12-23 13:02:16 +05:30
Igor Filippov e8d15b22d1 [Pipeline] animatediff + vid2vid + controlnet (#9337)
* add animatediff + vid2vide + controlnet

* post tests fixes

* PR discussion fixes

* update docs

* change input video to links on HF + update an example

* make quality fix

* fix ip adapter test

* fix ip adapter test input

* update ip adapter test
2024-12-23 13:02:16 +05:30
YiYi Xu 04809b0ccb refactor get_timesteps for SDXL img2img + add set_begin_index (#9375)
* refator + add begin_index

* add kolors img2img to doc
2024-12-23 13:02:16 +05:30
Sayak Paul b1571b6382 modify benchmarks to replace sdv1.5 with dreamshaper. (#9334) 2024-12-23 13:02:16 +05:30
YiYi Xu d5a529773a add flux inpaint + img2img + controlnet to auto pipeline (#9367) 2024-12-23 13:02:16 +05:30
Dhruv Nair 5af81b7ce4 [CI] Quick fix for Cog Video Test (#9373)
update
2024-12-23 13:02:16 +05:30
Aryan 402f547d78 [core] Freenoise memory improvements (#9262)
* update

* implement prompt interpolation

* make style

* resnet memory optimizations

* more memory optimizations; todo: refactor

* update

* update animatediff controlnet with latest changes

* refactor chunked inference changes

* remove print statements

* update

* chunk -> split

* remove changes from incorrect conflict resolution

* remove changes from incorrect conflict resolution

* add explanation of SplitInferenceModule

* update docs

* Revert "update docs"

This reverts commit c55a50a271.

* update docstring for freenoise split inference

* apply suggestions from review

* add tests

* apply suggestions from review
2024-12-23 13:02:16 +05:30
Haruya Ishikawa e48213cdc8 fix one uncaught deprecation warning for accessing vae_latent_channels in VaeImagePreprocessor (#9372)
deprecation warning vae_latent_channels
2024-12-23 13:02:16 +05:30
Linoy Tsaban 8e5940ea62 improve README for flux dreambooth lora (#9290)
* improve readme

* improve readme

* improve readme

* improve readme
2024-12-23 13:02:16 +05:30
Dhruv Nair 5c02d93aac [CI] Update Single file Nightly Tests (#9357)
* update

* update
2024-12-23 13:02:16 +05:30
Tolga Cangöz b4ab5b0ed7 Update UNet2DConditionModel's error messages (#9230)
* refactor
2024-12-23 13:02:16 +05:30
Vishnu V Jaddipal 2a72888dc0 Add Flux inpainting and Flux Img2Img (#9135)
---------

Co-authored-by: yiyixuxu <yixu310@gmail.com>
2024-12-23 13:02:16 +05:30
Fanli Lin 704226ef68 [tests] make 2 tests device-agnostic (#9347)
* enabel on xpu

* fix style
2024-12-23 13:02:16 +05:30
Eduardo Escobar 58f598160f Enable load_lora_weights for StableDiffusion3InpaintPipeline (#9330)
Enable load_lora_weights for StableDiffusion3InpaintPipeline

Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
2024-12-23 13:02:16 +05:30
YiYi Xu ce9b8ac30b small update on rotary embedding (#9354)
* update

* fix

---------

Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
2024-12-23 13:02:16 +05:30
Sayak Paul 32f85ed014 chore: add a cleaning utility to be useful during training. (#9240) 2024-12-23 13:02:16 +05:30
Aryan c636d53456 [refactor] move positional embeddings to patch embed layer for CogVideoX (#9263)
* remove frame limit in cogvideox

* remove debug prints

* Update src/diffusers/models/transformers/cogvideox_transformer_3d.py

* revert pipeline; remove frame limitation

* revert transformer changes

* address review comments

* add error message

* apply suggestions from review
2024-12-23 13:02:16 +05:30
Dhruv Nair f16a47f8b0 [CI] Add option to dispatch Fast GPU tests on main (#9355)
update
2024-12-23 13:02:16 +05:30
Aryan 7dc5fd05f2 [tests] remove/speedup some low signal tests (#9285)
* remove 2 shapes from SDFunctionTesterMixin::test_vae_tiling

* combine freeu enable/disable test to reduce many inference runs

* remove low signal unet test for signature

* remove low signal embeddings test

* remove low signal progress bar test from PipelineTesterMixin

* combine ip-adapter single and multi tests to save many inferences

* fix broken tests

* Update tests/pipelines/test_pipelines_common.py

* Update tests/pipelines/test_pipelines_common.py

* add progress bar tests
2024-12-23 13:02:16 +05:30
Dhruv Nair 0354a45f47 [CI] More Fast GPU Test Fixes (#9346)
* update

* update

* update

* update
2024-12-23 13:02:16 +05:30
Vishnu V Jaddipal 12d1663e4f Xlabs lora fix (#9348)
* Fix ```from_single_file``` for xl_inpaint

* Add basic flux inpaint pipeline

* style, quality, stray print

* Fix stray changes

* Add inpainting model support

* Change lora conversion for xlabs

* Fix stray changes

* Apply suggestions from code review

* style

---------

Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
2024-12-23 13:02:16 +05:30
Dhruv Nair d171e6a5ea [CI] More fixes for Fast GPU Tests on main (#9300)
update
2024-12-23 13:02:16 +05:30
Aryan 91168b5436 [core] Support VideoToVideo with CogVideoX (#9333)
* add vid2vid pipeline for cogvideox

* make fix-copies

* update docs

* fake context parallel cache, vae encode tiling

* add test for cog vid2vid

* use video link from HF docs repo

* add copied from comments; correctly rename test class
2024-12-23 13:02:16 +05:30
Aryan c466106eb8 [core] CogVideoX memory optimizations in VAE encode (#9340)
fake context parallel cache, vae encode tiling

(cherry picked from commit bf890bca0e)
2024-12-23 13:02:16 +05:30
YiYi Xu 4d901f9029 update runway repo for single_file (#9323)
update to a place holder
2024-12-23 13:02:15 +05:30
Aryan ae69462e0e [docs] Add a note on torchao/quanto benchmarks for CogVideoX and memory-efficient inference (#9296)
* add a note on torchao/quanto benchmarks and memory-efficient inference

* apply suggestions from review

* update

* Update docs/source/en/api/pipelines/cogvideox.md

Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>

* Update docs/source/en/api/pipelines/cogvideox.md

Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>

* add note on enable sequential cpu offload

---------

Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>
2024-12-23 13:02:15 +05:30
Dhruv Nair b6cd4a4337 [CI] Update Hub Token on nightly tests (#9318)
update
2024-12-23 13:02:15 +05:30
YiYi Xu ae8d9bb77a refactor rotary embedding 3: so it is not on cpu (#9307)
change get_1d_rotary to accept pos as torch tensors
2024-12-23 13:02:15 +05:30
YiYi Xu 17b838f5ae rotary embedding refactor 2: update comments, fix dtype for use_real=False (#9312)
fix notes and dtype
2024-12-23 13:02:15 +05:30
Anand Kumar fdb1759b48 [train_custom_diffusion.py] Fix the LR schedulers when num_train_epochs is passed in a distributed training env (#9308)
* Update train_custom_diffusion.py to fix the LR schedulers for `num_train_epochs`

* Fix saving text embeddings during safe serialization

* Fixed formatting
2024-12-23 13:02:15 +05:30
Sayak Paul 348578e4bc [LoRA] support kohya and xlabs loras for flux. (#9295)
* support kohya lora in flux.

* format

* support xlabs

* diffusion_model prefix.

* Apply suggestions from code review

Co-authored-by: apolinário <joaopaulo.passos@gmail.com>

* empty commit.

Co-authored-by: Leommm-byte <leom20031@gmail.com>

---------

Co-authored-by: apolinário <joaopaulo.passos@gmail.com>
Co-authored-by: Leommm-byte <leom20031@gmail.com>
2024-12-23 13:02:15 +05:30
apolinário 7cfc8bd488 Change default for guidance_scalein FLUX (#9305)
To match the original code, 7.0 is too high
2024-12-23 13:02:15 +05:30
Aryan 34b6db7d24 AnimateDiff prompt travel (#9231)
* update

* implement prompt interpolation

* make style

* resnet memory optimizations

* more memory optimizations; todo: refactor

* update

* update animatediff controlnet with latest changes

* refactor chunked inference changes

* remove print statements

* undo memory optimization changes

* update docstrings

* fix tests

* fix pia tests

* apply suggestions from review

* add tests

* update comment
2024-12-23 13:02:15 +05:30
Frank (Haofan) Wang b71158d2ec Update __init__.py (#9286) 2024-12-23 13:02:15 +05:30
Dhruv Nair 53721b9758 Fix Flux CLIP prompt embeds repeat for num_images_per_prompt > 1 (#9280)
update
2024-12-23 13:02:15 +05:30
Dhruv Nair 76c402e233 [CI] Update Release Tests (#9274)
* update

* update
2024-12-23 13:02:15 +05:30
Dhruv Nair dc014c46e3 Fix Freenoise for AnimateDiff V3 checkpoint. (#9288)
update
2024-12-23 13:02:15 +05:30
Marçal Comajoan Cara 433dce672b Replace transformers.deepspeed with transformers.integrations.deepspeed (#9281)
to avoid "FutureWarning: transformers.deepspeed module is deprecated and will be removed in a future version. Please import deepspeed modules directly from transformers.integrations"

Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
2024-12-23 13:02:15 +05:30