Update ldm/modules/encoders/modules.py

ldm/modules/encoders/modules.py

@@ -54,7 +54,7 @@ def disabled_train(self, mode=True):
 class FrozenFLANEmbedder(AbstractEncoder):
     """Uses the T5 transformer encoder for text"""
 
-    def __init__(self, version="google/flan-t5-large", device="cpu", max_length=77,
+    def __init__(self, version="google/flan-t5-large", device="cuda", max_length=77,
                  freeze=True):  # others are google/t5-v1_1-xl and google/t5-v1_1-xxl
         super().__init__()