search

LAION-AI / dalle2-laion

Pretrained Dalle2 from laion
500 stars 65 forks source link

Cuda:1 doesnt work #27

Open KyriaAnnwyn opened 2 years ago

KyriaAnnwyn commented 2 years ago

If I set "devices": "cuda:1" in config json, the model is still loaded to device 0 and the error that tensors are on different devices in calculations occurs.

nousr commented 1 year ago

feel free to open a PR with a fix if you want?