From 5bcf85a532df1de380b1a3377db2e5aaefaca1dd Mon Sep 17 00:00:00 2001 From: Haoming Date: Fri, 29 May 2026 10:10:36 +0800 Subject: [PATCH] emph --- backend/text_processing/anima_engine.py | 2 +- backend/text_processing/classic_engine.py | 2 +- backend/text_processing/gemma_engine.py | 2 +- backend/text_processing/klein_engine.py | 2 +- backend/text_processing/ministral3_engine.py | 2 +- backend/text_processing/parsing.py | 2 +- backend/text_processing/qwen3_engine.py | 2 +- backend/text_processing/qwen_engine.py | 2 +- backend/text_processing/t5_engine.py | 2 +- backend/text_processing/umt5_engine.py | 2 +- 10 files changed, 10 insertions(+), 10 deletions(-) diff --git a/backend/text_processing/anima_engine.py b/backend/text_processing/anima_engine.py index 9de1e845..df74fb35 100644 --- a/backend/text_processing/anima_engine.py +++ b/backend/text_processing/anima_engine.py @@ -21,7 +21,7 @@ class PromptChunk: class AnimaTextProcessingEngine: def __init__(self, text_encoder, qwen_tokenizer, t5_tokenizer): - super().__init__() + self.emphasis = emphasis.get_current_option(opts.emphasis)() self.text_encoder: "Qwen3_06B" = text_encoder self.qwen_tokenizer = qwen_tokenizer diff --git a/backend/text_processing/classic_engine.py b/backend/text_processing/classic_engine.py index bb31999a..04d5d206 100644 --- a/backend/text_processing/classic_engine.py +++ b/backend/text_processing/classic_engine.py @@ -51,7 +51,7 @@ class CLIPEmbeddingForTextualInversion(torch.nn.Module): class ClassicTextProcessingEngine: def __init__(self, text_encoder, tokenizer, chunk_length=75, embedding_dir=None, embedding_key="clip_l", embedding_expected_shape=768, text_projection=False, minimal_clip_skip=1, clip_skip=1, return_pooled=False, final_layer_norm=True): - super().__init__() + self.emphasis = emphasis.get_current_option(opts.emphasis)() self.embeddings = EmbeddingDatabase(tokenizer, embedding_expected_shape) diff --git a/backend/text_processing/gemma_engine.py b/backend/text_processing/gemma_engine.py index 8fffdf83..d75944a9 100644 --- a/backend/text_processing/gemma_engine.py +++ b/backend/text_processing/gemma_engine.py @@ -22,7 +22,7 @@ class PromptChunk: class GemmaTextProcessingEngine: def __init__(self, text_encoder, tokenizer): - super().__init__() + self.emphasis = emphasis.get_current_option(opts.emphasis)() self.text_encoder = text_encoder self.tokenizer = tokenizer diff --git a/backend/text_processing/klein_engine.py b/backend/text_processing/klein_engine.py index e604c68b..7ca9da78 100644 --- a/backend/text_processing/klein_engine.py +++ b/backend/text_processing/klein_engine.py @@ -22,7 +22,7 @@ class PromptChunk: class KleinTextProcessingEngine: def __init__(self, text_encoder, tokenizer): - super().__init__() + self.emphasis = emphasis.get_current_option(opts.emphasis)() self.text_encoder = text_encoder self.tokenizer = tokenizer diff --git a/backend/text_processing/ministral3_engine.py b/backend/text_processing/ministral3_engine.py index 7f3e150e..dac45a7a 100644 --- a/backend/text_processing/ministral3_engine.py +++ b/backend/text_processing/ministral3_engine.py @@ -16,7 +16,7 @@ class PromptChunk: class Ministral3TextProcessingEngine: def __init__(self, text_encoder, tokenizer): - super().__init__() + self.emphasis = emphasis.EmphasisNone() self.text_encoder = text_encoder self.tokenizer = tokenizer diff --git a/backend/text_processing/parsing.py b/backend/text_processing/parsing.py index 995f11e0..2d8b2455 100644 --- a/backend/text_processing/parsing.py +++ b/backend/text_processing/parsing.py @@ -22,7 +22,7 @@ re_attention = re.compile( re_break = re.compile(r"\s*\bBREAK\b\s*", re.S) -def parse_prompt_attention(text, emphasis): +def parse_prompt_attention(text: str, emphasis: str): res = [] round_brackets = [] square_brackets = [] diff --git a/backend/text_processing/qwen3_engine.py b/backend/text_processing/qwen3_engine.py index 41a1d0ad..3f1b8f65 100644 --- a/backend/text_processing/qwen3_engine.py +++ b/backend/text_processing/qwen3_engine.py @@ -22,7 +22,7 @@ class PromptChunk: class Qwen3TextProcessingEngine: def __init__(self, text_encoder, tokenizer): - super().__init__() + self.emphasis = emphasis.get_current_option(opts.emphasis)() self.text_encoder = text_encoder self.tokenizer = tokenizer diff --git a/backend/text_processing/qwen_engine.py b/backend/text_processing/qwen_engine.py index f8df2502..6e2370f4 100644 --- a/backend/text_processing/qwen_engine.py +++ b/backend/text_processing/qwen_engine.py @@ -17,7 +17,7 @@ class PromptChunk: class QwenTextProcessingEngine: def __init__(self, text_encoder, tokenizer): - super().__init__() + self.emphasis = emphasis.get_current_option(opts.emphasis)() self.text_encoder = text_encoder self.tokenizer = tokenizer diff --git a/backend/text_processing/t5_engine.py b/backend/text_processing/t5_engine.py index d9f90540..806b99ad 100644 --- a/backend/text_processing/t5_engine.py +++ b/backend/text_processing/t5_engine.py @@ -14,7 +14,7 @@ class PromptChunk: class T5TextProcessingEngine: def __init__(self, text_encoder, tokenizer, min_length: int = 256, min_padding: int = -1): - super().__init__() + self.emphasis = emphasis.get_current_option(opts.emphasis)() self.text_encoder = text_encoder.transformer self.tokenizer = tokenizer diff --git a/backend/text_processing/umt5_engine.py b/backend/text_processing/umt5_engine.py index 7b2d8266..f424d619 100644 --- a/backend/text_processing/umt5_engine.py +++ b/backend/text_processing/umt5_engine.py @@ -17,7 +17,7 @@ class PromptChunk: class UMT5TextProcessingEngine: def __init__(self, text_encoder, tokenizer, min_length=512): - super().__init__() + self.emphasis = emphasis.get_current_option(opts.emphasis)() self.text_encoder = text_encoder.transformer self.tokenizer = tokenizer