huggingface / huggingface/diffusers

Dreambooth Flux with Lora: OOM

Ouverte
#10,145 13 commentaires 0 réactions 0 personnes assignées Voir sur GitHub
bug stale training
Langage dominant
Python
Étoiles
34.5k
Forks
7.3k
Merge moyen
3 j 3 h
PR mergées (30 j)
91

Description

### 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_

Guide de contribution

Ouvrir le guide de contribution

Piste de recherche

Aucun fichier, test, journal ni étape de reproduction exploitable n’est fourni. Commencez par reproduire l’entraînement DreamBooth Flux LoRA avec le matériel L40S et les paramètres indiqués, puis identifiez l’échec lié à la mémoire et vérifiez un correctif en terminant l’entraînement sur une image sans erreurs d’out-of-memory.

Rédigé par le modèle d'indexation à partir du texte de l'issue.

Évaluation

Stack technique
python, pytorch
Domaine
machine-learning
Type d'issue
Bug
Difficulté
4/5
Temps estimé
3-5 jours
Activité
À l'abandon
Clarté
À clarifier
Accessibilité débutants
20/100

Recevez les nouvelles issues par e-mail

Un résumé court des issues GitHub adaptées aux débutants.