huggingface / huggingface/diffusers

Why set_lora_device doesn't work

Offen
#9,913 8 Kommentare 0 Reaktionen 0 zugewiesene Personen Auf GitHub ansehen
bug needs-code-example stale
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

Beitragsleitfaden öffnen

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

Neue Issues direkt in Ihr Postfach

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