huggingface / huggingface/diffusers

Allow user to pick which Text Encoders to train with SD3-Medium.

Open
#8,504 2 comments 0 reactions 0 assignees View on GitHub
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.**
Kinda yes, it seems both training scripts for LoRA and Dreambooth load all 3 Text Encoders and Tokenizers which leads to high vram usage, 20GB on Rank 32 Lora, batch size 1, with AdamW8Bit

**Describe the solution you'd like.**
Allow users to train the Text Encoder(s) they want.

**Additional context.**
![imagen](https://github.com/huggingface/diffusers/assets/23042093/4e868144-2abd-43a4-9070-e68c0bb52e52)

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.