Subject-Diffusion
Subject-Diffusion copied to clipboard
Why model is not in cuda?
Thanks for your great work and sharing your code !!!
When i run the code , model is in cpu , not in GPU how to solve it? I mantually add the code "self..xx.cuda()" , but it is not working!
Looking forward to your reply!