Skip to content

GitLab

Commit a5991a7a authored by comfyanonymous's avatar comfyanonymous
Browse files

Fix hunyuan dit text encoder weights always being in fp32.

parent 2c038cce
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
comfy/text_encoders/hydit.py
View file @ a5991a7a
...@@ -52,8 +52,8 @@ class HyditTokenizer: ...@@ -52,8 +52,8 @@ class HyditTokenizer:
class HyditModel(torch.nn.Module): class HyditModel(torch.nn.Module):
def __init__(self, device="cpu", dtype=None): def __init__(self, device="cpu", dtype=None):
super().__init__() super().__init__()
self.hydit_clip = HyditBertModel() self.hydit_clip = HyditBertModel(dtype=dtype)
self.mt5xl = MT5XLModel() self.mt5xl = MT5XLModel(dtype=dtype)
self.dtypes = set() self.dtypes = set()
if dtype is not None: if dtype is not None:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment