huggingface / huggingface/diffusers
Dreambooth Flux with Lora: OOM
- Lenguaje dominante
- Python
- Estrellas
- 34.5k
- Forks
- 7.3k
- Merge medio
- 3 d 3 h
- PR fusionados (30 d)
- 91
Descripción
### Describe the bug
It is pretty wired that even after I set lora rank to only 2, training the dreambooth_flux with lora still returns OOM. My GPU is three L40s (44G-48G). And data is only one image.
If this is still OOM, I'm wondering why should lora version exist? The next GPU memory capbility is 80G which can train it without lora.
### Reproduction
NA
### Logs
_No response_
### System Info
L40s
### Who can help?
_No response_
Guía de contribución
Línea de trabajo
No files, tests, logs, or usable reproduction steps are provided. Start by reproducing DreamBooth Flux LoRA training with the stated L40S hardware and settings, then identify the memory failure and verify a fix by completing the one-image training run without out-of-memory errors.
Escrito por el modelo de indexación a partir del texto del issue.
Evaluación
- Stack tecnológico
- python, pytorch
- Área
- machine-learning
- Tipo de issue
- Error
- Dificultad
- 4/5
- Tiempo estimado
- 3-5 días
- Estado de actividad
- Estancado
- Claridad
- Necesita aclaración
- Aptitud para principiantes
- 20/100