Update default device setting to CPU in encoder modules for improved compatibility across systems without GPU support.

imagedream/ldm/modules/encoders/modules.py

@@ -62,7 +62,7 @@ class FrozenT5Embedder(AbstractEncoder):
     """Uses the T5 transformer encoder for text"""
 
     def __init__(
-        self, version="google/t5-v1_1-large", device="cuda", max_length=77, freeze=True
+        self, version="google/t5-v1_1-large", device="cpu", max_length=77, freeze=True
     ):  # others are google/t5-v1_1-xl and google/t5-v1_1-xxl
         super().__init__()
         self.tokenizer = T5Tokenizer.from_pretrained(version)
@@ -106,7 +106,7 @@ class FrozenCLIPEmbedder(AbstractEncoder):
     def __init__(
         self,
         version="openai/clip-vit-large-patch14",
-        device="cuda",
+        device="cpu",
         max_length=77,
         freeze=True,
         layer="last",
@@ -173,7 +173,7 @@ class FrozenOpenCLIPEmbedder(AbstractEncoder, nn.Module):
         self,
         arch="ViT-H-14",
         version="laion2b_s32b_b79k",
-        device="cuda",
+        device="cpu",
         max_length=77,
         freeze=True,
         layer="last",
@@ -306,7 +306,7 @@ class FrozenCLIPT5Encoder(AbstractEncoder):
         self,
         clip_version="openai/clip-vit-large-patch14",
         t5_version="google/t5-v1_1-xl",
-        device="cuda",
+        device="cpu",
         clip_max_length=77,
         t5_max_length=77,
     ):