mirror of
https://github.com/jags111/efficiency-nodes-comfyui.git
synced 2026-03-21 13:12:13 -03:00
Update bnk_adv_encode.py
This commit is contained in:
committed by
GitHub
parent
1f02149480
commit
f6d01dc544
@@ -223,7 +223,7 @@ def encode_token_weights_l(model, token_weight_pairs):
|
||||
|
||||
def encode_token_weights(model, token_weight_pairs, encode_func):
|
||||
if model.layer_idx is not None:
|
||||
model.cond_stage_model.clip_layer(model.layer_idx)
|
||||
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)
|
||||
|
||||
Reference in New Issue
Block a user