Fix hunyuan dit text encoder weights always being in fp32.

a5991a7a · comfyanonymous · 2c038cce · a5991a7a
Commit a5991a7a authored Jul 31, 2024 by comfyanonymous
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

comfy/text_encoders/hydit.py comfy/text_encoders/hydit.py +2 -2

No files found.
--- a/comfy/text_encoders/hydit.py
+++ b/comfy/text_encoders/hydit.py
@@ -52,8 +52,8 @@ class HyditTokenizer:
 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: