alphard/efficiency-nodes-comfyui
1
0
Fork 0
mirror of https://github.com/jags111/efficiency-nodes-comfyui.git synced 2026-03-21 21:22:13 -03:00
Code Issues Packages Projects Releases Wiki Activity

Update bnk_adv_encode.py

Browse Source
This commit is contained in:
VALADI K JAGANATHAN
2023-11-01 11:55:12 +05:30
committed by GitHub
parent c3409c19bd
commit e2718b4fc7
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
py/bnk_adv_encode.py
View File

@@ -235,6 +235,7 @@ def prepareXL(embs_l, embs_g, pooled, clip_balance):
return torch.cat([embs_l * l_w, embs_g * g_w], dim=-1), pooled
else:
return embs_g, pooled
#=====================
def advanced_encode(clip, text, token_normalization, weight_interpretation, w_max=1.0, clip_balance=.5, apply_to_pooled=True):
tokenized = clip.tokenize(text, return_word_ids=True)
@@ -271,7 +272,8 @@ def advanced_encode(clip, text, token_normalization, weight_interpretation, w_ma
weight_interpretation,
lambda x: (clip.encode_from_tokens(x), None),
w_max=w_max)
#=====================
def advanced_encode_XL(clip, text1, text2, token_normalization, weight_interpretation, w_max=1.0, clip_balance=.5, apply_to_pooled=True):
tokenized1 = clip.tokenize(text1, return_word_ids=True)
tokenized2 = clip.tokenize(text2, return_word_ids=True)