wuyingwen/comfyui
1
0
Fork 0
Code Issues Pull Requests Actions 5 Packages Projects Releases Wiki Activity

Fix hunyuan dit text encoder weights always being in fp32.

Browse Source
This commit is contained in:
comfyanonymous 2024-07-31 01:34:57 -04:00
parent 2c038ccef0
commit a5991a7aa6
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/text_encoders/hydit.py
View File

@ -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: