lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-20 14:29:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Apply emphasis mode changes immediately (#2423)

Browse Source 
Previously emphasis mode changes applied only on model load.
Also added handling for mode 'None' (literal interpretation), which previously gave results identical to mode 'Ignore'.
This commit is contained in:
DenOfEquity
2024-12-08 19:25:39 +00:00
committed by GitHub
parent 2e867aaf3a
commit 2051c4100b
3 changed files with 51 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
backend/text_processing/classic_engine.py
View File

@@ -6,6 +6,8 @@ from backend.text_processing import parsing, emphasis
from backend.text_processing.textual_inversion import EmbeddingDatabase
from backend import memory_management
from modules.shared import opts
PromptChunkFix = namedtuple('PromptChunkFix', ['offset', 'embedding'])
last_extra_generation_params = {}
@@ -67,7 +69,7 @@ class ClassicTextProcessingEngine:
self.text_encoder = text_encoder
self.tokenizer = tokenizer
self.emphasis = emphasis.get_current_option(emphasis_name)()
self.emphasis = emphasis.get_current_option(opts.emphasis)()
self.text_projection = text_projection
self.minimal_clip_skip = minimal_clip_skip
self.clip_skip = clip_skip
@@ -146,7 +148,7 @@ class ClassicTextProcessingEngine:
return z
def tokenize_line(self, line):
parsed = parsing.parse_prompt_attention(line)
parsed = parsing.parse_prompt_attention(line, self.emphasis.name)
tokenized = self.tokenize([text for text, _ in parsed])
@@ -248,6 +250,8 @@ class ClassicTextProcessingEngine:
return batch_chunks, token_count
def __call__(self, texts):
self.emphasis = emphasis.get_current_option(opts.emphasis)()
batch_chunks, token_count = self.process_texts(texts)
used_embeddings = {}