Aryan
0e1c569c58
update tests
2024-09-17 21:52:33 +02:00
Aryan
0c8ec36353
remove option to train text encoder
...
Co-Authored-By: bghira <bghira@users.github.com >
2024-09-17 21:51:08 +02:00
Aryan
57d7ca647e
update docs
2024-09-17 03:51:20 +02:00
Aryan
f07755fd04
import Dummy optim and scheduler only wheh required
2024-09-17 03:50:57 +02:00
Aryan
6d704ce770
fight lora runner tests
2024-09-17 03:20:36 +02:00
Aryan
2ec6232d1c
Merge branch 'main' into cogvideox-lora-and-training
2024-09-17 02:17:20 +02:00
Linoy Tsaban
8fcfb2a456
[Flux with CFG] add flux pipeline with cfg support ( #9445 )
...
* true_cfg
* add check negative prompt/embeds inputs
* move to community pipelines
* move to community pipelines
* revert true cfg changes to the orig pipline
* style
---------
Co-authored-by: YiYi Xu <yixu310@gmail.com >
2024-09-16 12:09:34 -10:00
suzukimain
b52119ae92
[docs] Replace runwayml/stable-diffusion-v1-5 with Lykon/dreamshaper-8 ( #9428 )
...
* [docs] Replace runwayml/stable-diffusion-v1-5 with Lykon/dreamshaper-8
Updated documentation as runwayml/stable-diffusion-v1-5 has been removed from Huggingface.
* Update docs/source/en/using-diffusers/inpaint.md
Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com >
* Replace with stable-diffusion-v1-5/stable-diffusion-v1-5
* Update inpaint.md
---------
Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com >
2024-09-16 10:18:45 -07:00
Aryan
ec8d483e72
deepspeed refactor
2024-09-16 13:54:34 +02:00
Aryan
b07ac7496a
add requirements.txt
2024-09-16 03:23:02 +02:00
Aryan
4679088bfe
deepspeed stuff
2024-09-16 03:05:29 +02:00
Aryan
a3ca2a2732
update lora tests
2024-09-16 02:57:10 +02:00
Aryan
e3a9fe7bde
Merge branch 'main' into cogvideox-lora-and-training
2024-09-15 22:25:25 +02:00
Aryan
e3c267750b
add test skeleton
2024-09-15 22:25:20 +02:00
Aryan
e83200ece5
update
2024-09-15 22:03:23 +02:00
Aryan
96b2f17b1a
fix
2024-09-15 20:27:44 +02:00
Linoy Tsaban
37e3603c4a
[Flux Dreambooth lora] add latent caching ( #9160 )
...
* add ostris trainer to README & add cache latents of vae
* add ostris trainer to README & add cache latents of vae
* style
* readme
* add test for latent caching
* add ostris noise scheduler
9ee1ef2a0a/toolkit/samplers/custom_flowmatch_sampler.py (L95)
* style
* fix import
* style
* fix tests
* style
* --change upcasting of transformer?
* update readme according to main
* keep only latent caching
* add configurable param for final saving of trained layers- --upcast_before_saving
* style
* Update examples/dreambooth/README_flux.md
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
* Update examples/dreambooth/README_flux.md
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
* use clear_objs_and_retain_memory from utilities
* style
---------
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
2024-09-15 15:30:31 +03:00
Aryan
0bd238a522
make style
2024-09-15 00:07:56 +02:00
Aryan
8aa62dd436
update
2024-09-15 00:07:29 +02:00
Aryan
f1740b0fb5
Merge branch 'main' into cogvideox-lora-and-training
2024-09-14 23:44:05 +02:00
Aryan
5e5ee430cb
update
2024-09-14 23:41:53 +02:00
Leo Jiang
e2ead7cdcc
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-09-14 16:29:38 +05:30
Aryan
e5a44fd55c
add README
2024-09-14 04:30:01 +02:00
Aryan
80c87718ad
apply suggestions from review; prodigy optimizer
...
YiYi Xu <yixu310@gmail.com >
2024-09-14 02:36:07 +02:00
Aryan
165d9b2dd9
Merge branch 'main' into cogvideox-lora-and-training
2024-09-12 13:04:54 +05:30
Aryan
7e9e25cf6c
Update examples/cogvideo/train_cogvideox_lora.py
...
Co-authored-by: YiYi Xu <yixu310@gmail.com >
2024-09-12 13:04:31 +05:30
Juan Acevedo
45aa8bb187
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-09-12 08:35:06 +05:30
Yu Zheng
c002731d93
[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-09-11 07:04:37 +05:30
Sayak Paul
adf1f911f0
[Tests] fix some fast gpu tests. ( #9379 )
...
fix some fast gpu tests.
2024-09-11 06:50:02 +05:30
Aryan
f35c36c16d
use correct lora scaling for final test pipeline
2024-09-09 23:45:47 +02:00
Aryan
b1ca3dbadf
fix lora alpha
2024-09-07 22:27:54 +02:00
Aryan
4159b3bc82
Update examples/cogvideo/train_cogvideox_lora.py
...
Co-authored-by: YiYi Xu <yixu310@gmail.com >
2024-09-08 00:08:25 +05:30
Aryan
a063503044
apply suggestions from review
2024-09-07 14:46:29 +02:00
Aryan
038fec4ac9
update
2024-09-07 13:25:18 +02:00
Aryan
3afb2e2540
Merge branch 'main' into cogvideox-lora-and-training
2024-09-06 23:04:50 +02:00
Linoy Tsaban
55ac421f7b
improve README for flux dreambooth lora ( #9290 )
...
* improve readme
* improve readme
* improve readme
* improve readme
2024-09-05 17:53:23 +05:30
Aryan
4c56287535
update
2024-09-04 07:58:17 +02:00
Sayak Paul
8ba90aa706
chore: add a cleaning utility to be useful during training. ( #9240 )
2024-09-03 15:00:17 +05:30
Aryan
9a95d8de56
update
2024-09-03 05:57:48 +02:00
Aryan
63e80b77fc
Merge branch 'main' into cogvideox-lora-and-training
2024-09-03 05:22:22 +02:00
Aryan
74e6f90097
update
2024-09-01 02:35:19 +02:00
Aryan
588c6ee602
update
2024-08-31 15:41:26 +02:00
Aryan
24c362ca4f
update
2024-08-29 18:53:04 +02:00
Anand Kumar
40c13fe5b4
[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-08-29 14:23:36 +05:30
Aryan
f12e669ed3
update
2024-08-28 15:51:57 +02:00
Aryan
dc08234a7c
cogvideox lora training draft
2024-08-28 15:35:30 +02:00
Linoy Tsaban
c977966502
[Dreambooth flux] bug fix for dreambooth script (align with dreambooth lora) ( #9257 )
...
* fix shape
* fix prompt encoding
* style
* fix device
* add comment
2024-08-26 17:29:58 +05:30
Leo Jiang
eda36c4c28
Fix dtype error for StableDiffusionXL ( #9217 )
...
Fix dtype error
Co-authored-by: 蒋硕 <jiangshuo9@h-partners.com >
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
2024-08-20 06:58:50 +05:30
Wenlong Wu
815d882217
Add loading text inversion ( #9130 )
2024-08-19 09:26:27 +05:30
townwish4git
d25eb5d385
fix(sd3): fix deletion of text_encoders etc ( #8951 )
...
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
Co-authored-by: YiYi Xu <yixu310@gmail.com >
2024-08-18 15:37:40 -10:00