nodes/city96-ComfyUI_ExtraModels
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

PixArt SD3 node prompt weight fix

Browse Source 
#61
This commit is contained in:
City
2024-06-16 02:51:21 +02:00
parent c174662c5b
commit 9a4629aba7
1 changed files with 15 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
PixArt/nodes.py
+14 -2
View File
@@ -214,13 +214,25 @@ class PixArtT5FromSD3CLIP:
def split(self, sd3_clip, padding): def split(self, sd3_clip, padding):
from comfy.sd3_clip import SD3Tokenizer, SD3ClipModel from comfy.sd3_clip import SD3Tokenizer, SD3ClipModel
import copy
clip = sd3_clip.clone() clip = sd3_clip.clone()
assert clip.cond_stage_model.t5xxl is not None, "CLIP must have T5 loaded!" assert clip.cond_stage_model.t5xxl is not None, "CLIP must have T5 loaded!"
# model # remove transformer
transformer = clip.cond_stage_model.t5xxl.transformer
clip.cond_stage_model.t5xxl.transformer = None
# clone object
tmp = SD3ClipModel(clip_l=False, clip_g=False, t5=False) tmp = SD3ClipModel(clip_l=False, clip_g=False, t5=False)
tmp.t5xxl = clip.cond_stage_model.t5xxl tmp.t5xxl = copy.deepcopy(clip.cond_stage_model.t5xxl)
# put transformer back
clip.cond_stage_model.t5xxl.transformer = transformer
tmp.t5xxl.transformer = transformer
# override special tokens
tmp.t5xxl.special_tokens = copy.deepcopy(clip.cond_stage_model.t5xxl.special_tokens)
tmp.t5xxl.special_tokens.pop("end") # make sure empty tokens match
# tokenizer # tokenizer
tok = SD3Tokenizer() tok = SD3Tokenizer()