Fix hunyuan dit text encoder weights always being in fp32.

NeoWorldTeam · Jul 31, 2024 · a5991a7 · a5991a7
1 parent 2c038cc
commit a5991a7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/comfy/text_encoders/hydit.py b/comfy/text_encoders/hydit.py
@@ -52,8 +52,8 @@ def state_dict(self):
 class HyditModel(torch.nn.Module):
     def __init__(self, device="cpu", dtype=None):
         super().__init__()
-        self.hydit_clip = HyditBertModel()
-        self.mt5xl = MT5XLModel()
+        self.hydit_clip = HyditBertModel(dtype=dtype)
+        self.mt5xl = MT5XLModel(dtype=dtype)
 
         self.dtypes = set()
         if dtype is not None: