vincedovy
a536e775fb
Fix json WindowsPath crash ( #8662 )
...
* Add check for WindowsPath in to_json_string
On Windows, os.path.join returns a WindowsPath. to_json_string does not convert this from a WindowsPath to a string. Added check for WindowsPath to to_json_saveable.
* Remove extraneous convert to string in test_check_path_types (tests/others/test_config.py)
* Fix style issues in tests/others/test_config.py
* Add unit test to test_config.py to verify that PosixPath and WindowsPath (depending on system) both work when converted to JSON
* Remove distinction between PosixPath and WindowsPath in ConfigMixIn.to_json_string(). Conditional now tests for Path, and uses Path.as_posix() to convert to string.
---------
Co-authored-by: Vincent Dovydaitis <vincedovy@gmail.com >
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
2024-06-27 10:30:55 +05:30
Γlvaro Somoza
3b01d72a64
Modify FlowMatch Scale Noise ( #8678 )
...
* initial fix
* apply suggestion
* delete step_index line
2024-06-27 00:36:33 -04:00
Sayak Paul
e2a4a46e99
[Release notification] add some info when there is an error. ( #8718 )
...
add some info when there is an error.
2024-06-27 09:49:15 +05:30
Sayak Paul
eda560d34c
modify PR and issue templates ( #8687 )
...
* modify PR and issue templates
* add single file poc.
2024-06-27 09:01:47 +05:30
Sayak Paul
adbb04864d
[LoRA] fix conversion utility so that lora dora loads correctly ( #8688 )
...
fix conversion utility so that lora dora loads correctly
2024-06-27 08:58:32 +05:30
Dhruv Nair
effe4b9784
Update xformers SD3 test ( #8712 )
...
update
2024-06-26 10:24:27 -10:00
Sayak Paul
5b51ad0052
[LoRA] fix vanilla fine-tuned lora loading. ( #8691 )
...
fix vanilla fine-tuned lora loading.
2024-06-26 07:38:57 -10:00
Sayak Paul
10b4e354b6
[Chore] remove deprecation from transformer2d regarding the output class. ( #8698 )
...
* remove deprecation from transformer2d regarding the output class.
* up
* deprecate more
2024-06-26 07:35:36 -10:00
Donald.Lee
ea6938aea5
Fix: unet save_attn_procs at UNet2DconditionLoadersMixin ( #8699 )
...
* fix: unet save_attn_procs at custom diffusion
* style: recover unchanaged parts(max line length 119) / mod: add condition
* style: recover unchanaged parts(max line length 119)
---------
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
2024-06-26 22:30:49 +05:30
Sayak Paul
8ef0d9deff
[Observability] add reporting mechanism when mirroring community pipelines. ( #8676 )
...
* add reporting mechanism when mirroring community pipelines.
* remove unneeded argument
* get the actual PATH_IN_REPO
* don't need tag
2024-06-26 22:11:33 +05:30
XCL
fa2abfdb03
[Tencent Hunyuan Team] Add Hunyuan-DiT ControlNet Inference ( #8694 )
...
* add controlnet support
---------
Co-authored-by: xingchaoliu <xingchaoliu@tencent.com >
Co-authored-by: yiyixuxu <yixu310@gmail,com>
2024-06-26 00:43:03 -10:00
YiYi Xu
1d3ef67b09
[doc] add more about from_pipe API for PAG doc ( #8701 )
...
* add more about from_pipe API
* Update docs/source/en/using-diffusers/pag.md
* Update docs/source/en/using-diffusers/pag.md
---------
Co-authored-by: yiyixuxu <yixu310@gmail,com>
2024-06-25 22:26:12 -10:00
Dhruv Nair
0f0b531827
Add decorator for compile tests ( #8703 )
...
* update
* update
---------
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
2024-06-26 11:26:47 +05:30
Sayak Paul
e8284281c1
add docs on model sharding ( #8658 )
...
* add docs on model sharding
* add entry to _toctree.
* Apply suggestions from code review
Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com >
* simplify wording
* add a note on transformer library handling
* move device placement section
* Update docs/source/en/training/distributed_inference.md
Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com >
---------
Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com >
2024-06-26 07:35:11 +05:30
YiYi Xu
715a7da1b2
add sd3 conversion script ( #8702 )
...
add conversion script
2024-06-25 14:24:58 -10:00
Γlvaro Somoza
14d224d4e6
[Docs] SD3 T5 Token limit doc ( #8654 )
...
* doc for max_sequence_length
* better position and changed note to tip
* apply suggestions
---------
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
2024-06-25 14:41:27 -04:00
YiYi Xu
540399f540
add PAG support ( #7944 )
...
* first draft
---------
Co-authored-by: yiyixuxu <yixu310@gmail,com>
Co-authored-by: Junhwa Song <ethan9867@gmail.com >
Co-authored-by: Ahn Donghoon (μλν / suno) <suno.vivid@gmail.com >
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com >
2024-06-25 08:40:02 -10:00
Sayak Paul
f088027e93
[Marigold tests] add is_flaky decorator to some Marigold tests ( #8696 )
...
okay
2024-06-25 06:27:28 -10:00
Linoy Tsaban
c6e08ecd46
[Sd3 Dreambooth LoRA] Add text encoder training for the clip encoders ( #8630 )
...
* add clip text-encoder training
* no dora
* text encoder traing fixes
* text encoder traing fixes
* text encoder training fixes
* text encoder training fixes
* text encoder training fixes
* text encoder training fixes
* add text_encoder layers to save_lora
* style
* fix imports
* style
* fix text encoder
* review changes
* review changes
* review changes
* minor change
* add lora tag
* style
* add readme notes
* add tests for clip encoders
* style
* typo
* fixes
* style
* Update tests/lora/test_lora_layers_sd3.py
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
* Update examples/dreambooth/README_sd3.md
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
* minor readme change
---------
Co-authored-by: YiYi Xu <yixu310@gmail.com >
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
2024-06-25 18:00:19 +05:30
Sayak Paul
4ad7a1f5fd
[Chore] create a utility for calculating the expected number of shards. ( #8692 )
...
create a utility for calculating the expected number of shards.
2024-06-25 17:05:39 +05:30
Hammond Liu
1f81fbe274
Fix redundant pipe init in sd3 lora ( #8680 )
...
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
2024-06-25 07:31:20 +05:30
Tolga CangΓΆz
589931ca79
Errata - Update class method convention to use cls ( #8574 )
...
* Class methods are supposed to use `cls` conventionally
* `make style && make quality`
* An Empty commit
---------
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
2024-06-24 10:35:45 -07:00
Steven Liu
675be88f00
[docs] Add note for float8 ( #8685 )
...
add note
2024-06-24 10:13:34 -07:00
Steven Liu
df4ad6f4ac
[docs] Fix Pillow import ( #8684 )
...
fix import error
2024-06-24 10:13:15 -07:00
Sayak Paul
bc90c28bc9
[Docs] add note on caching in fast diffusion ( #8675 )
...
* add note on caching in fast diffusion
* formatting
* Update docs/source/en/tutorials/fast_diffusion.md
Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com >
---------
Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com >
2024-06-24 10:10:45 -07:00
Tolga CangΓΆz
f040c27d4c
Errata - Fix typos and improve style ( #8571 )
...
* Fix typos
* Fix typos & up style
* chore: Update numbers
---------
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
2024-06-24 10:07:22 -07:00
Tolga CangΓΆz
138fac703a
Discourage using deprecated revision parameter ( #8573 )
...
* Discourage using `revision`
* `make style && make quality`
* Refactor code to use 'variant' instead of 'revision'
* `revision="bf16"` -> `variant="bf16"`
---------
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
2024-06-24 10:06:49 -07:00
Tolga CangΓΆz
468ae09ed8
Errata - Trim trailing white space in the whole repo ( #8575 )
...
* Trim all the trailing white space in the whole repo
* Remove unnecessary empty places
* make style && make quality
* Trim trailing white space
* trim
---------
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
2024-06-24 18:39:15 +05:30
Dong
3fca52022f
π¨ fix xl playground device ( #8550 )
...
* π¨ fix xl playground device
* π¨ run `make fix-copies`
* π¨ run `make fix-copies`
* edit xl_controlnet_img2img file
* edit playground img2img test slow
* Update tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl_img2img.py
---------
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
2024-06-24 16:49:55 +05:30
Tolga CangΓΆz
c375903db5
Errata - Fix typos & improve contributing page ( #8572 )
...
* Fix typos & improve contributing page
* `make style && make quality`
* fix typos
* Fix typo
---------
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
2024-06-24 14:13:03 +05:30
Vinh H. Pham
b9d52fca1d
[train_lcm_distill_lora_sdxl.py] Fix the LR schedulers when num_train_epochs is passed in a distributed training env ( #8446 )
...
fix num_train_epochs
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
2024-06-24 14:09:28 +05:30
drhead
2ada094bff
Add extra performance features for EMAModel, torch._foreach operations and better support for non-blocking CPU offloading ( #7685 )
...
* Add support for _foreach operations and non-blocking to EMAModel
* default foreach to false
* add non-blocking EMA offloading to SD1.5 T2I example script
* fix whitespace
* move foreach to cli argument
* linting
* Update README.md re: EMA weight training
* correct args.foreach_ema
* add tests for foreach ema
* code quality
* add foreach to from_pretrained
* default foreach false
* fix linting
---------
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
Co-authored-by: drhead <a@a.a>
2024-06-24 14:03:47 +05:30
Haofan Wang
f1f542bdd4
Update pipeline_stable_diffusion_3_controlnet.py ( #8660 )
...
Co-authored-by: YiYi Xu <yixu310@gmail,com>
2024-06-23 15:27:59 +05:30
Sayak Paul
a9c403c001
[LoRA] refactor lora conversion utility. ( #8295 )
...
* refactor lora conversion utility.
* remove error raises.
* add onetrainer support too.
2024-06-22 08:29:12 +05:30
Γlvaro Somoza
e7b9a0762b
[SD3 LoRA] Fix list index out of range ( #8584 )
...
* fix
* add check
* key present is checked before
* test case draft
* aply suggestions
* changed testing repo, back to old class
* forgot docstring
---------
Co-authored-by: YiYi Xu <yixu310@gmail.com >
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
2024-06-21 21:17:34 +05:30
Sayak Paul
8eb17315c8
[LoRA] get rid of the legacy lora remnants and make our codebase lighter ( #8623 )
...
* get rid of the legacy lora remnants and make our codebase lighter
* fix depcrecated lora argument
* fix
* empty commit to trigger ci
* remove print
* empty
2024-06-21 16:36:05 +05:30
YiYi Xu
c71c19c5e6
a few fix for shard checkpoints ( #8656 )
...
fix
Co-authored-by: yiyixuxu <yixu310@gmail,com>
2024-06-21 12:50:58 +05:30
Steaunk
adc31940a9
Fix Typo in StableDiffusion3 ( #8642 )
...
* fix typo in __call__ of pipeline_stable_diffusion_3.py
* fix typo in __call__ of pipeline_stable_diffusion_3_img2img.py
---------
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
2024-06-21 08:45:48 +05:30
satani99
963ee05d16
Update train_dreambooth_lora_sd3.py ( #8600 )
...
* Update train_dreambooth_lora_sd3.py
* Update train_dreambooth_lora_sd3.py
* Update train_dreambooth_sd3.py
---------
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
2024-06-20 17:42:24 +05:30
Sayak Paul
668e34c6e0
[LoRA SD3] add support for lora fusion in sd3 ( #8616 )
...
* add support for lora fusion in sd3
* add test to ensure fused lora and effective lora produce same outpouts
2024-06-20 14:25:51 +05:30
Sayak Paul
25d7bb3ea6
[Flax tests] reduce tolerance for a flax test ( #8640 )
...
reduce tolerance for a flax test
2024-06-20 00:48:08 +04:00
YiYi Xu
394b8fb996
fix from_single_file for checkpoints with t5 ( #8631 )
...
fix single file
2024-06-19 08:23:35 -10:00
Sayak Paul
a1d55e14ba
Change the default weighting_scheme in the SD3 scripts ( #8639 )
...
* change to logit_normal as the weighting scheme
* sensible default mote
2024-06-19 13:05:26 +01:00
ηε₯ε
e5564d45bf
Support SD3 ControlNet and Multi-ControlNet. ( #8566 )
...
* sd3 controlnet
---------
Co-authored-by: haofanwang <haofanwang.ai@gmail.com >
2024-06-18 14:59:22 -10:00
Nan
2921a20194
[SD3] Fix mis-matched shape when num_images_per_prompt > 1 using without T5 (text_encoder_3=None) ( #8558 )
...
* fix shape mismatch when num_images_per_prompt > 1 and text_encoder_3=None
* style
* fix copies
---------
Co-authored-by: YiYi Xu <yixu310@gmail.com >
Co-authored-by: yiyixuxu <yixu310@gmail,com>
2024-06-18 12:41:18 -10:00
Carolinabanana
3376252d71
Fix gradient checkpointing issue for Stable Diffusion 3 ( #8542 )
...
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
Co-authored-by: YiYi Xu <yixu310@gmail.com >
2024-06-18 11:36:23 -10:00
Yongsen Mao
16170c69ae
add sd1.5 compatibility to controlnet-xs and fix unused_parameters error during training ( #8606 )
...
* add sd1.5 compatibility to controlnet-xs
* set use_linear_projection by base_block
* refine code style
2024-06-18 11:35:34 -10:00
kkj15dk
4408047ac5
self.upsample = Upsample1D ( #8580 )
...
Making self.upsample actually be Upsample1D
2024-06-18 11:34:07 -10:00
Vasco Ramos
34fab8b511
[SD3 Docs] Corrected title about loading model with T5 "without" -> "with" ( #8602 )
...
[SD3 Docs] Corrected title about loading model with T5
Corrected the documentation title to "Loading the single file checkpoint with T5" Previously, it incorrectly stated "Loading the single file checkpoint without T5" which contradicted the code snippet showing how to load the SD3 checkpoint with the T5 model
2024-06-18 11:33:43 -10:00
Gæros
298ce67999
[LoRA] text encoder: read the ranks for all the attn modules ( #8324 )
...
* [LoRA] text encoder: read the ranks for all the attn modules
* In addition to out_proj, read the ranks of adapters for q_proj, k_proj, and v_proj
* Allow missing adapters (UNet already supports this)
* ruff format loaders.lora
* [LoRA] add tests for partial text encoders LoRAs
* [LoRA] update test_simple_inference_with_partial_text_lora to be deterministic
* [LoRA] comment justifying test_simple_inference_with_partial_text_lora
* style
---------
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
2024-06-18 21:10:50 +01:00