cloneofsimo / lora

Using Low-rank adaptation to quickly fine-tune diffusion models.
https://arxiv.org/abs/2106.09685
Apache License 2.0
6.99k stars 481 forks source link

Can I use multiple GPUs for training? #224

Open pf67 opened 1 year ago

pf67 commented 1 year ago

set --Device = "cuda: 0"\ seems to only specify one gpu