ai-forever / ai-forever/Kandinsky-2
Использует память GPU , когда явно указано 'cpu'
Aperta
- Lingua principale
- Jupyter Notebook
- Stelle
- 2.8k
- Fork
- 319
- Metriche di merge delle PR
- Nessuna PR unita negli ultimi 30g
Descrizione
Code to reproduce:
```
from kandinsky2 import get_kandinsky2
model = get_kandinsky2('cpu', task_type='text2img', cache_dir='/tmp/kandinsky2', model_version='2.1', use_flash_attention=False)
```
after run `nvidia-smi`
Guida per i contributori
Nessuna guida per i contributori indicizzata per questo repository
Valutazione
Questa issue non è ancora stata valutata.