mirror of
https://github.com/wassname/Clover-Edition.git
synced 2026-09-10 11:40:48 +08:00
fixed mental issue?
This commit is contained in:
@@ -168,6 +168,8 @@ class CTRLGenerator():
|
||||
def result_replace(self, result):
|
||||
print("\n\nBEFORE RESULT_REPLACE:")
|
||||
print(repr(result))
|
||||
|
||||
first_letter_capitalized = result[0].isupper()
|
||||
result = result.replace("#", "")
|
||||
result = first_to_second_person(result)
|
||||
result = remove_profanity(result)
|
||||
@@ -175,6 +177,9 @@ class CTRLGenerator():
|
||||
print("\n\nAFTER RESULT_REPLACE:")
|
||||
print(repr(result))
|
||||
|
||||
if not first_letter_capitalized:
|
||||
result[0] = result[0].lower()
|
||||
|
||||
return result
|
||||
|
||||
def generate(self, prompt, options=None):
|
||||
@@ -285,4 +290,5 @@ class CTRLGenerator():
|
||||
result = tokens_generated_so_far[prompt_length:]
|
||||
first_token = False
|
||||
|
||||
|
||||
return self.result_replace(result)
|
||||
|
||||
@@ -61,6 +61,8 @@ def cut_trailing_sentence(text):
|
||||
|
||||
def replace_outside_quotes(text, current_word, repl_word):
|
||||
|
||||
current_word = current_word.replace(".", "\.")
|
||||
|
||||
reg_expr = re.compile(current_word + '(?=([^"]*"[^"]*")*[^"]*$)')
|
||||
|
||||
output = reg_expr.sub(repl_word, text)
|
||||
|
||||
Reference in New Issue
Block a user