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

Set t5_attention_mask for new comfy versions

Browse Source 
#79
This commit is contained in:
City
2024-10-24 21:00:57 +02:00
parent 193c2fd9d3
commit c6527a45f1
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
PixArt/nodes.py
+4
View File
@@ -253,6 +253,10 @@ class PixArtT5FromSD3CLIP:
tmp.t5xxl.special_tokens = copy.deepcopy(clip.cond_stage_model.t5xxl.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 tmp.t5xxl.special_tokens.pop("end") # make sure empty tokens match
# add attn mask opt if present in original
if hasattr(sd3_clip.cond_stage_model, "t5_attention_mask"):
tmp.t5_attention_mask = False
# tokenizer # tokenizer
tok = SD3Tokenizer() tok = SD3Tokenizer()
tok.t5xxl.min_length = padding tok.t5xxl.min_length = padding