cuda

ldm/guided_diffusion/loss_vq.py

@@ -54,7 +54,7 @@ class VQLPIPSWithDiscriminator(nn.Module):
         self.pixel_weight = pixelloss_weight
         if perceptual_loss == "lpips":
             print(f"{self.__class__.__name__}: Running with LPIPS.")
-            self.perceptual_loss = LPIPS().eval().to(device="cuda")
+            self.perceptual_loss = LPIPS().eval().cuda()
         else:
             raise ValueError(f"Unknown perceptual loss: >> {perceptual_loss} <<")
         self.perceptual_weight = perceptual_weight