|
from modules import sd_hijack_clip |
|
from modules import shared |
|
|
|
|
|
def process_text_old(self: sd_hijack_clip.FrozenCLIPEmbedderWithCustomWordsBase, texts): |
|
id_start = self.id_start |
|
id_end = self.id_end |
|
maxlen = self.wrapped.max_length |
|
used_custom_terms = [] |
|
remade_batch_tokens = [] |
|
hijack_comments = [] |
|
hijack_fixes = [] |
|
token_count = 0 |
|
|
|
cache = {} |
|
batch_tokens = self.tokenize(texts) |
|
batch_multipliers = [] |
|
for tokens in batch_tokens: |
|
tuple_tokens = tuple(tokens) |
|
|
|
if tuple_tokens in cache: |
|
remade_tokens, fixes, multipliers = cache[tuple_tokens] |
|
else: |
|
fixes = [] |
|
remade_tokens = [] |
|
multipliers = [] |
|
mult = 1.0 |
|
|
|
i = 0 |
|
while i < len(tokens): |
|
token = tokens[i] |
|
|
|
embedding, embedding_length_in_tokens = self.hijack.embedding_db.find_embedding_at_position(tokens, i) |
|
|
|
mult_change = self.token_mults.get(token) if shared.opts.enable_emphasis else None |
|
if mult_change is not None: |
|
mult *= mult_change |
|
i += 1 |
|
elif embedding is None: |
|
remade_tokens.append(token) |
|
multipliers.append(mult) |
|
i += 1 |
|
else: |
|
emb_len = int(embedding.vec.shape[0]) |
|
fixes.append((len(remade_tokens), embedding)) |
|
remade_tokens += [0] * emb_len |
|
multipliers += [mult] * emb_len |
|
used_custom_terms.append((embedding.name, embedding.checksum())) |
|
i += embedding_length_in_tokens |
|
|
|
if len(remade_tokens) > maxlen - 2: |
|
vocab = {v: k for k, v in self.wrapped.tokenizer.get_vocab().items()} |
|
ovf = remade_tokens[maxlen - 2:] |
|
overflowing_words = [vocab.get(int(x), "") for x in ovf] |
|
overflowing_text = self.wrapped.tokenizer.convert_tokens_to_string(''.join(overflowing_words)) |
|
hijack_comments.append(f"Warning: too many input tokens; some ({len(overflowing_words)}) have been truncated:\n{overflowing_text}\n") |
|
|
|
token_count = len(remade_tokens) |
|
remade_tokens = remade_tokens + [id_end] * (maxlen - 2 - len(remade_tokens)) |
|
remade_tokens = [id_start] + remade_tokens[0:maxlen - 2] + [id_end] |
|
cache[tuple_tokens] = (remade_tokens, fixes, multipliers) |
|
|
|
multipliers = multipliers + [1.0] * (maxlen - 2 - len(multipliers)) |
|
multipliers = [1.0] + multipliers[0:maxlen - 2] + [1.0] |
|
|
|
remade_batch_tokens.append(remade_tokens) |
|
hijack_fixes.append(fixes) |
|
batch_multipliers.append(multipliers) |
|
return batch_multipliers, remade_batch_tokens, used_custom_terms, hijack_comments, hijack_fixes, token_count |
|
|
|
|
|
def forward_old(self: sd_hijack_clip.FrozenCLIPEmbedderWithCustomWordsBase, texts): |
|
batch_multipliers, remade_batch_tokens, used_custom_terms, hijack_comments, hijack_fixes, token_count = process_text_old(self, texts) |
|
|
|
self.hijack.comments += hijack_comments |
|
|
|
if len(used_custom_terms) > 0: |
|
self.hijack.comments.append("Used embeddings: " + ", ".join([f'{word} [{checksum}]' for word, checksum in used_custom_terms])) |
|
|
|
self.hijack.fixes = hijack_fixes |
|
return self.process_tokens(remade_batch_tokens, batch_multipliers) |
|
|
