diff --git a/fooocus_version.py b/fooocus_version.py index 543540dd..299cfc77 100644 --- a/fooocus_version.py +++ b/fooocus_version.py @@ -1 +1 @@ -version = '1.0.51' +version = '1.0.52' diff --git a/modules/expansion.py b/modules/expansion.py index 391f5694..32daab9e 100644 --- a/modules/expansion.py +++ b/modules/expansion.py @@ -3,6 +3,10 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline, set_seed from modules.path import fooocus_expansion_path +def safe_str(x): + return str(x).rstrip(",. \r\n") + + class FooocusExpansion: def __init__(self): self.tokenizer = AutoTokenizer.from_pretrained(fooocus_expansion_path) @@ -15,9 +19,10 @@ class FooocusExpansion: print('Fooocus Expansion engine loaded.') def __call__(self, prompt, seed): - prompt = str(prompt).rstrip('\n') + prompt = safe_str(prompt) + '. ' # reduce semantic corruption. seed = int(seed) set_seed(seed) response = self.pipe(prompt, max_length=len(prompt) + 256) - result = response[0]['generated_text'].rstrip('\n') + result = response[0]['generated_text'] + result = safe_str(result) return result diff --git a/update_log.md b/update_log.md index 6b63138f..4985055f 100644 --- a/update_log.md +++ b/update_log.md @@ -1,3 +1,7 @@ +### 1.0.52 + +* Reduce the semantic corruption of Prompt Expansion. + ### 1.0.51 * Speed up Prompt Expansion a bit.