Update ldm/modules/encoders/modules.py

ldm/modules/encoders/modules.py

@@ -144,7 +144,7 @@ class FrozenOpenCLIPEmbedder(AbstractEncoder):
                  freeze=True, layer="last"):
         super().__init__()
         assert layer in self.LAYERS
-        model, _, _ = open_clip.create_model_and_transforms(arch, device=torch.device('cpu'), pretrained=version)
+        model, _, _ = open_clip.create_model_and_transforms(arch, device=torch.device('cuda'), pretrained=version)
         del model.visual
         self.model = model