Subject-Diffusion icon indicating copy to clipboard operation
Subject-Diffusion copied to clipboard

Why model is not in cuda?

Open Ed-ivan opened this issue 2 years ago • 0 comments

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!

Ed-ivan avatar Oct 20 '23 14:10 Ed-ivan