wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-21 20:00:17 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

Fix clip clone bug

Browse Source
This commit is contained in:
doctorpangloss 2024-05-17 07:17:33 -07:00
parent b318b4cc28
commit 4eb66f8a0a
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
comfy/sdxl_clip.py
View File

@ -1,3 +1,5 @@
import copy
import torch import torch
from . import sd1_clip from . import sd1_clip
@ -37,6 +39,12 @@ class SDXLTokenizer:
def untokenize(self, token_weight_pair): def untokenize(self, token_weight_pair):
return self.clip_g.untokenize(token_weight_pair) return self.clip_g.untokenize(token_weight_pair)
def clone(self):
sdxl_tokenizer = copy.copy(self)
sdxl_tokenizer.clip_l = self.clip_l.clone()
sdxl_tokenizer.clip_g = self.clip_g.clone()
return sdxl_tokenizer
class SDXLClipModel(torch.nn.Module): class SDXLClipModel(torch.nn.Module):
def __init__(self, device="cpu", dtype=None): def __init__(self, device="cpu", dtype=None):