From f6d01dc5449e2cc6e29da92b3b337e49c0dce112 Mon Sep 17 00:00:00 2001 From: VALADI K JAGANATHAN Date: Tue, 27 Feb 2024 07:16:44 +0530 Subject: [PATCH] Update bnk_adv_encode.py --- py/bnk_adv_encode.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/py/bnk_adv_encode.py b/py/bnk_adv_encode.py index 8464726..b3d251f 100644 --- a/py/bnk_adv_encode.py +++ b/py/bnk_adv_encode.py @@ -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)