Let tokenizers return weights to be stored in the saved checkpoint.

This commit is contained in:
comfyanonymous
2024-07-25 10:52:09 -04:00
parent 10c919f4c7
commit f87810cd3e
4 changed files with 15 additions and 1 deletions

View File

@@ -34,6 +34,9 @@ class SDXLTokenizer:
def untokenize(self, token_weight_pair):
return self.clip_g.untokenize(token_weight_pair)
def state_dict(self):
return {}
class SDXLClipModel(torch.nn.Module):
def __init__(self, device="cpu", dtype=None):
super().__init__()