Commit Graph

4527 Commits

Author SHA1 Message Date
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
Steven Liu 4fc5e29ab0 [docs] Add pipelines to table (#9282)
update pipelines
2024-12-23 13:02:15 +05:30
Álvaro Somoza 3f17b8d56e [IP Adapter] Fix cache_dir and local_files_only for image encoder (#9272)
initial fix
2024-12-23 13:02:15 +05:30
Linoy Tsaban 18a3c35aac [Dreambooth flux] bug fix for dreambooth script (align with dreambooth lora) (#9257)
* fix shape

* fix prompt encoding

* style

* fix device

* add comment
2024-12-23 13:02:15 +05:30
YiYi Xu 9c185d8d84 refactor 3d rope for cogvideox (#9269)
* refactor 3d rope

* repeat -> expand
2024-12-23 13:02:15 +05:30
王奇勋 cfd12e2ece [Flux] Support Union ControlNet (#9175)
* refactor
---------

Co-authored-by: haofanwang <haofanwang.ai@gmail.com>
2024-12-23 13:02:15 +05:30
yangpei-comp 5a9de1993c Bugfix in pipeline_kandinsky2_2_combined.py: Image type check mismatch (#9256)
Update pipeline_kandinsky2_2_combined.py

Bugfix on image type check mismatch
2024-12-23 13:02:15 +05:30
Dhruv Nair a2462e3e3a [CI] Run Fast + Fast GPU Tests on release branches. (#9255)
* update

* update
2024-12-23 13:02:15 +05:30
Dhruv Nair 47ab1f5089 [Single File] Add Flux Pipeline Support (#9244)
update
2024-12-23 13:02:15 +05:30
Dhruv Nair 9483ab4040 [Single File] Support loading Comfy UI Flux checkpoints (#9243)
update
2024-12-23 13:02:15 +05:30
Sayak Paul f24f48340f [Core] fuse_qkv_projection() to Flux (#9185)
* start fusing flux.

* test

* finish fusion

* fix-copues
2024-12-23 13:02:15 +05:30
zR 977afbe83b Cogvideox-5B Model adapter change (#9203)
* draft of embedding

---------

Co-authored-by: Aryan <aryan@huggingface.co>
2024-12-23 13:02:15 +05:30
YiYi Xu 51317a8ecd fix _identify_model_variants (#9247)
merge now, will add/fix test next
2024-12-23 13:02:15 +05:30
Elias Rad de303b9a15 Docs fix spelling issues (#9219)
* fix PHILOSOPHY.md

* fix CONTRIBUTING.md

* fix tutorial_overview.md

* fix stable_diffusion.md

* Update tutorial_overview.md
2024-12-23 13:02:15 +05:30