cpu -> cuda

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("cpu"),
+            device=torch.device("cuda"),
             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("cpu"), pretrained=version
+            arch, device=torch.device("cuda"), 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("cpu"),
+            device=torch.device("cuda"),
             pretrained=version,
         )
         del model.transformer