mirror of
https://github.com/jags111/efficiency-nodes-comfyui.git
synced 2026-03-21 21:22: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):
|
def encode_token_weights(model, token_weight_pairs, encode_func):
|
||||||
if model.layer_idx is not None:
|
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)
|
model_management.load_model_gpu(model.patcher)
|
||||||
return encode_func(model.cond_stage_model, token_weight_pairs)
|
return encode_func(model.cond_stage_model, token_weight_pairs)
|
||||||
|
|||||||
Reference in New Issue
Block a user