Update bnk_adv_encode.py

This commit is contained in:
VALADI K JAGANATHAN
2023-11-01 11:49:37 +05:30
committed by GitHub
parent cc54d1d5df
commit c3409c19bd

View File

@@ -238,7 +238,7 @@ def prepareXL(embs_l, embs_g, pooled, clip_balance):
def advanced_encode(clip, text, token_normalization, weight_interpretation, w_max=1.0, clip_balance=.5, apply_to_pooled=True):
tokenized = clip.tokenize(text, return_word_ids=True)
if isinstance(tokenized, dict):
if isinstance(clip.cond_stage_model, (SDXLClipModel, SDXLRefinerClipModel, SDXLClipG)):
embs_l = None
embs_g = None
pooled = None