diff --git a/modules/expansion.py b/modules/expansion.py index 1595e91f..8ff038b8 100644 --- a/modules/expansion.py +++ b/modules/expansion.py @@ -3,6 +3,15 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline, set_seed from modules.path import fooocus_expansion_path +magic_split = [ + ', extremely', + ', by', + ', trending', + ', best', + '. ', +] + + def safe_str(x): x = str(x) for _ in range(16): @@ -22,9 +31,11 @@ class FooocusExpansion: print('Fooocus Expansion engine loaded.') def __call__(self, prompt, seed): - prompt = safe_str(prompt) + '. ' # Reduce semantic corruption. seed = int(seed) set_seed(seed) + + prompt = safe_str(prompt) + magic_split[seed % len(magic_split)] + response = self.pipe(prompt, max_length=len(prompt) + 256) result = response[0]['generated_text'] result = safe_str(result) diff --git a/update_log.md b/update_log.md index cb8a7232..3ba794a0 100644 --- a/update_log.md +++ b/update_log.md @@ -1,5 +1,9 @@ ### 1.0.55 +* Begin to use magic split. + +### 1.0.55 + * Minor changes of Prompt Expansion. ### 1.0.52