huggingface / huggingface/diffusers
How to finetune Flux-dev full params, 80G OOM ...
Offen
#10,196
21 Kommentare
1 Reaktion
1 zugewiesene Person
Beansprucht von @linoytsaban
Auf GitHub ansehen
training
- Vorherrschende Sprache
- Python
- Sterne
- 34.5k
- Forks
- 7.3k
- Ø Merge
- 3 T. 3 Std.
- Gemergte PRs (30 T.)
- 91
Beschreibung
I am using the [train_dreambooth_flux](https://github.com/huggingface/diffusers/blob/main/examples/dreambooth/train_dreambooth_flux.py) script to fine-tune the `flux-dev` model with full parameters using DeepSpeed Stage 2. However, I am still encountering out-of-memory issues on an 80GB GPU. Are there any solutions available to address this problem? Thanks!
Beitragsleitfaden
Bewertung
Dieses Issue wurde noch nicht bewertet.