3 Commits

Author	SHA1	Message	Date
Linoy Tsaban	ba31a14700	[SD 3.5 Dreambooth LoRA] support configurable training block & layers (#9762) ... * configurable layers * configurable layers * update README * style * add test * style * add layer test, update readme, add nargs * readme * test style * remove print, change nargs * test arg change * style * revert nargs 2/2 * address sayaks comments * style * address sayaks comments	2024-12-23 13:02:17 +05:30
Linoy Tsaban	85dde53e30	[SD3 dreambooth-lora training] small updates + bug fixes (#9682) ... * add latent caching + smol updates * update license * replace with free_memory * add --upcast_before_saving to allow saving transformer weights in lower precision * fix models to accumulate * fix mixed precision issue as proposed in https://github.com/huggingface/diffusers/pull/9565 * smol update to readme * style * fix caching latents * style * add tests for latent caching * style * fix latent caching --------- Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>	2024-12-23 13:02:17 +05:30
Sayak Paul	f5f9f9c083	[Tests] add test suite for SD3 DreamBooth (#8650) ... * add a test suite for SD3 DreamBooth * lora suite * style * add checkpointing tests for LoRA * add test to cover train_text_encoder.	2024-12-23 13:02:13 +05:30