stablediffusion icon indicating copy to clipboard operation
stablediffusion copied to clipboard

Update depth2img.py

Open mzamini92 opened this issue 1 year ago • 0 comments

In the paint() function, you are using the torch.autocast() function to cast the batch tensor to the cuda device. However, the batch tensor is already on the cuda device, so there is no need to cast it again. This can be fixed by removing the torch.autocast() call from the paint() function. Am I right?

mzamini92 avatar Jul 11 '23 14:07 mzamini92