Rollback to cpu

sgm/modules/encoders/modules.py

@@ -529,7 +529,7 @@ class FrozenOpenCLIPEmbedder2(AbstractEmbModel):
         assert layer in self.LAYERS
         model, _, _ = open_clip.create_model_and_transforms(
             arch,
-            device=torch.device("cuda"),
+            device=torch.device("cpu"),
             pretrained=version if SDXL_CLIP2_CKPT_PTH is None else SDXL_CLIP2_CKPT_PTH,
         )
         del model.visual
@@ -628,7 +628,7 @@ class FrozenOpenCLIPEmbedder(AbstractEmbModel):
         super().__init__()
         assert layer in self.LAYERS
         model, _, _ = open_clip.create_model_and_transforms(
-            arch, device=torch.device("cuda"), pretrained=version
+            arch, device=torch.device("cpu"), pretrained=version
         )
         del model.visual
         self.model = model
@@ -703,7 +703,7 @@ class FrozenOpenCLIPImageEmbedder(AbstractEmbModel):
         super().__init__()
         model, _, _ = open_clip.create_model_and_transforms(
             arch,
-            device=torch.device("cuda"),
+            device=torch.device("cpu"),
             pretrained=version,
         )
         del model.transformer
@@ -1059,4 +1059,4 @@ class GaussianEncoder(Encoder, AbstractEmbModel):
         log["weight"] = self.weight
         if self.flatten_output:
             z = rearrange(z, "b c h w -> b (h w ) c")
-        return log, z
+        return log, z