From 740b3f920f02c8e79d7a6a36fafffe87e5614600 Mon Sep 17 00:00:00 2001 From: VALADI K JAGANATHAN Date: Tue, 31 Oct 2023 09:53:01 +0530 Subject: [PATCH] Update bnk_adv_encode.py changes to how advanced encode work .. possibly resolves the token issues --- py/bnk_adv_encode.py | 21 ++++++++++++++------- 1 file changed, 14 insertions(+), 7 deletions(-) diff --git a/py/bnk_adv_encode.py b/py/bnk_adv_encode.py index 72d4548..0e30a05 100644 --- a/py/bnk_adv_encode.py +++ b/py/bnk_adv_encode.py @@ -242,13 +242,20 @@ def advanced_encode(clip, text, token_normalization, weight_interpretation, w_ma embs_l = None embs_g = None pooled = None - if 'l' in tokenized and isinstance(clip.cond_stage_model, SDXLClipModel): - embs_l, _ = advanced_encode_from_tokens(tokenized['l'], - token_normalization, - weight_interpretation, - lambda x: encode_token_weights(clip, x, encode_token_weights_l), - w_max=w_max, - return_pooled=False) + if 'l' in tokenized: + if isinstance(clip.cond_stage_model, SDXLClipModel): + embs_l, _ = advanced_encode_from_tokens(tokenized['l'], + token_normalization, + weight_interpretation, + lambda x: encode_token_weights(clip, x, encode_token_weights_l), + w_max=w_max, + return_pooled=False) + else: + return advanced_encode_from_tokens(tokenized['l'], + token_normalization, + weight_interpretation, + lambda x: encode_token_weights(clip, x, encode_token_weights_l), + w_max=w_max) if 'g' in tokenized: embs_g, pooled = advanced_encode_from_tokens(tokenized['g'], token_normalization,