Update ldm/modules/encoders/modules.py

ldm/modules/encoders/modules.py

@@ -103,7 +103,7 @@ class FrozenCLAPEmbedder(AbstractEncoder):
         self.caption_encoder = TextEncoder(
             args.d_proj, args.text_model, args.transformer_embed_dim
         )
-
+        device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
         self.max_length = max_length
         self.device = device
         if freeze: self.freeze()