Foreign-Mirrors/Fooocus
1
0
Fork 0
mirror of https://github.com/lllyasviel/Fooocus.git synced 2026-08-16 13:13:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

add logo

Browse Source
This commit is contained in:
lvmin
2023-09-10 12:03:25 -07:00
parent eeddd25ece
commit 5b111f09cc
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/expansion.py
+2 -2
View File
@@ -3,7 +3,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline, set_seed
from modules.path import fooocus_expansion_path from modules.path import fooocus_expansion_path
magic_split = [ fooocus_magic_split = [
', extremely', ', extremely',
', trending', ', trending',
', best', ', best',
@@ -36,7 +36,7 @@ class FooocusExpansion:
seed = int(seed) seed = int(seed)
set_seed(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) response = self.pipe(prompt, max_length=len(prompt) + 256)
result = response[0]['generated_text'] result = response[0]['generated_text']