diff --git a/comfy/text_encoders/hydit.py b/comfy/text_encoders/hydit.py index ac0c893b..9dfa288b 100644 --- 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: