huggingface / huggingface/diffusers

why rebuild a vae in inference stage?

Offen
#9,172 2 Kommentare 1 Reaktion 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

Thanks for ur effort for diffusion model.

I want to know why we need to rebuild a vae in inference stage. I think it will introduce extra GPU cost.
https://github.com/huggingface/diffusers/blob/a85b34e7fdc0a5fceb11aa0fa6199bd9afaca396/examples/text_to_image/train_text_to_image_sdxl.py#L1217C16-L1223C24

Beitragsleitfaden

Beitragsleitfaden öffnen

Rechercherichtung

Start at examples/text_to_image/train_text_to_image_sdxl.py around lines 1217-1223 and trace how the VAE is constructed during inference. Compare that lifecycle with the training setup and document whether the rebuild is required and what the GPU cost is.

Vom Indexierungsmodell aus dem Issue-Text verfasst.

Bewertung

Tech-Stack
python, pytorch
Bereich
machine-learning
Issue-Typ
Dokumentation
Schwierigkeit
3/5
Geschätzter Aufwand
1-2 Tage
Aktivitätsstatus
Veraltet
Klarheit
Muss geklärt werden
Anfängerfreundlichkeit
25/100

Neue Issues direkt in Ihr Postfach

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