浏览代码

Merge pull request #14203 from AUTOMATIC1111/remove-clean_text()

remove clean_text()
AUTOMATIC1111 1 年之前
父节点
当前提交
f92d61497a
共有 1 个文件被更改,包括 3 次插入20 次删除
  1. 3 20
      modules/styles.py

+ 3 - 20
modules/styles.py

@@ -2,7 +2,6 @@ import csv
 import fnmatch
 import fnmatch
 import os
 import os
 import os.path
 import os.path
-import re
 import typing
 import typing
 import shutil
 import shutil
 
 
@@ -14,22 +13,6 @@ class PromptStyle(typing.NamedTuple):
     path: str = None
     path: str = None
 
 
 
 
-def clean_text(text: str) -> str:
-    """
-    Iterating through a list of regular expressions and replacement strings, we
-    clean up the prompt and style text to make it easier to match against each
-    other.
-    """
-    re_list = [
-        ("multiple commas", re.compile("(,+\s+)+,?"), ", "),
-        ("multiple spaces", re.compile("\s{2,}"), " "),
-    ]
-    for _, regex, replace in re_list:
-        text = regex.sub(replace, text)
-
-    return text.strip(", ")
-
-
 def merge_prompts(style_prompt: str, prompt: str) -> str:
 def merge_prompts(style_prompt: str, prompt: str) -> str:
     if "{prompt}" in style_prompt:
     if "{prompt}" in style_prompt:
         res = style_prompt.replace("{prompt}", prompt)
         res = style_prompt.replace("{prompt}", prompt)
@@ -44,7 +27,7 @@ def apply_styles_to_prompt(prompt, styles):
     for style in styles:
     for style in styles:
         prompt = merge_prompts(style, prompt)
         prompt = merge_prompts(style, prompt)
 
 
-    return clean_text(prompt)
+    return prompt
 
 
 
 
 def unwrap_style_text_from_prompt(style_text, prompt):
 def unwrap_style_text_from_prompt(style_text, prompt):
@@ -56,8 +39,8 @@ def unwrap_style_text_from_prompt(style_text, prompt):
     Note that the "cleaned" version of the style text is only used for matching
     Note that the "cleaned" version of the style text is only used for matching
     purposes here. It isn't returned; the original style text is not modified.
     purposes here. It isn't returned; the original style text is not modified.
     """
     """
-    stripped_prompt = clean_text(prompt)
-    stripped_style_text = clean_text(style_text)
+    stripped_prompt = prompt
+    stripped_style_text = style_text
     if "{prompt}" in stripped_style_text:
     if "{prompt}" in stripped_style_text:
         # Work out whether the prompt is wrapped in the style text. If so, we
         # Work out whether the prompt is wrapped in the style text. If so, we
         # return True and the "inner" prompt text that isn't part of the style.
         # return True and the "inner" prompt text that isn't part of the style.