huggingface / huggingface/diffusers
SD3 dreambooth training - caching of embeddings and vae representations
Open
stale
- Dominant language
- Python
- Stars
- 34.5k
- Forks
- 7.3k
- Avg merge
- 3d 3h
- Merged PRs (30d)
- 91
Description
**Is your feature request related to a problem? Please describe.**
The current implementation of the dreambooth trainings for both loras and finetuning is very memory intensive.
**Describe the solution you'd like.**
I would like the option to pre-cache the VAE representations of the images and the text encoder representations so that training could be done without the text encoders being in VRAM.
**Describe alternatives you've considered.**
I don't believe there are alternatives to reducing the VRAM requirement for training.
Contributor guide
Assessment
This issue has not been assessed yet.