From 5b111f09cc2e70982cec1e6eb7008cecade2f414 Mon Sep 17 00:00:00 2001 From: lvmin Date: Sun, 10 Sep 2023 12:03:25 -0700 Subject: [PATCH] add logo --- modules/expansion.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modules/expansion.py b/modules/expansion.py index c3744e3f..36f454f0 100644 --- a/modules/expansion.py +++ b/modules/expansion.py @@ -3,7 +3,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline, set_seed from modules.path import fooocus_expansion_path -magic_split = [ +fooocus_magic_split = [ ', extremely', ', trending', ', best', @@ -36,7 +36,7 @@ class FooocusExpansion: seed = int(seed) set_seed(seed) - prompt = safe_str(prompt) + magic_split[seed % len(magic_split)] + prompt = safe_str(prompt) + fooocus_magic_split[seed % len(fooocus_magic_split)] response = self.pipe(prompt, max_length=len(prompt) + 256) result = response[0]['generated_text']