alphard/efficiency-nodes-comfyui
1
0
Fork 0
mirror of https://github.com/jags111/efficiency-nodes-comfyui.git synced 2026-05-06 16:56:44 -03:00
Code Issues Packages Projects Releases Wiki Activity

优化encode_token_weights函数以保持CLIP选项与ComfyUI核心编码路径一致 (#377)

Browse Source 
seems a additional update
This commit is contained in:
izum
2026-04-01 07:42:18 +08:00
committed by GitHub
parent cc91287d72
commit 65b2c0361f
1 changed files with 10 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
py/bnk_adv_encode.py
View File

@@ -221,12 +221,16 @@ def encode_token_weights_l(model, token_weight_pairs):
l_out, _ = model.clip_l.encode_token_weights(token_weight_pairs)
return l_out, None
def encode_token_weights(model, token_weight_pairs, encode_func):
if model.layer_idx is not None:
model.cond_stage_model.set_clip_options({"layer": model.layer_idx})
model_management.load_model_gpu(model.patcher)
return encode_func(model.cond_stage_model, token_weight_pairs)
def encode_token_weights(model, token_weight_pairs, encode_func):
# Keep CLIP options aligned with ComfyUI's core encode path so token
# tensors are created on the same device as the active text encoder pass.
model.cond_stage_model.reset_clip_options()
if model.layer_idx is not None:
model.cond_stage_model.set_clip_options({"layer": model.layer_idx})
model_management.load_model_gpu(model.patcher)
model.cond_stage_model.set_clip_options({"execution_device": model.patcher.load_device})
return encode_func(model.cond_stage_model, token_weight_pairs)
def prepareXL(embs_l, embs_g, pooled, clip_balance):
l_w = 1 - max(0, clip_balance - .5) * 2