huggingface / huggingface/diffusers

SD3 dreambooth training - caching of embeddings and vae representations

Offen
#8,540 2 Kommentare 0 Reaktionen 0 zugewiesene Personen Auf GitHub ansehen
stale
Vorherrschende Sprache
Python
Sterne
34.5k
Forks
7.3k
Ø Merge
3 T. 3 Std.
Gemergte PRs (30 T.)
91

Beschreibung

**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.

Beitragsleitfaden

Beitragsleitfaden öffnen

Rechercherichtung

Start by locating the SD3 DreamBooth training entry points and how they currently encode images and text during training. The requested result is optional pre-caching of VAE and text-encoder representations so training can run without the text encoders in VRAM; the issue names no specific files or tests.

Vom Indexierungsmodell aus dem Issue-Text verfasst.

Bewertung

Tech-Stack
python, pytorch
Bereich
machine-learning
Issue-Typ
Feature
Schwierigkeit
4/5
Geschätzter Aufwand
3-5 Tage
Aktivitätsstatus
Veraltet
Klarheit
Größtenteils klar
Anfängerfreundlichkeit
35/100

Neue Issues direkt in Ihr Postfach

Eine kurze Übersicht über anfängerfreundliche GitHub-Issues.