huggingface / huggingface/diffusers
Why set_lora_device doesn't work
- Vorherrschende Sprache
- Python
- Sterne
- 34.5k
- Forks
- 7.3k
- Ø Merge
- 3 T. 3 Std.
- Gemergte PRs (30 T.)
- 91
Beschreibung
### Describe the bug
When I load serverl loras with set_lora_device(), the GPU memory continues to grow, cames from 20G to 25G, this function doesn't work
### Reproduction
for key in lora_list:
weight_name = key + ".safetensors"
pipe.load_lora_weights(lora_path, weight_name=weight_name, adapter_name=key, local_files_only=True)
adapters = pipe.get_list_adapters()
print(adapters)
pipe.set_lora_device([key], torch.device('cpu'))
### Logs
_No response_
### System Info
V100 32G
diffusers 0.32.0.dev0
torch 2.0.1+cu118
peft 0.12.0
### Who can help?
_No response_
Beitragsleitfaden
Rechercherichtung
Beginnen Sie am Einstiegspunkt set_lora_device() und reproduzieren Sie die Schleife mit den gemeldeten Versionen von diffusers, torch und peft. Vergleichen Sie den GPU-Speicher nach jedem Laden und jeder Geräteverschiebung; abgeschlossen ist die Untersuchung, wenn das gemeldete Wachstum erklärt ist und das Problem entweder mit einem verifizierten Fix behoben oder eine dokumentierte Einschränkung festgestellt wurde.
Vom Indexierungsmodell aus dem Issue-Text verfasst.
Bewertung
- Tech-Stack
- python, pytorch
- Bereich
- machine-learning
- Issue-Typ
- Bug
- Schwierigkeit
- 4/5
- Geschätzter Aufwand
- 3-5 Tage
- Aktivitätsstatus
- Veraltet
- Klarheit
- Muss geklärt werden
- Anfängerfreundlichkeit
- 35/100