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

suffix mode

Browse Source
This commit is contained in:
lvmin
2023-09-10 17:48:03 -07:00
parent 9e9821b3ed
commit d50714b261
6 changed files with 19 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/expansion.py
+3 -4
View File
@@ -33,10 +33,9 @@ class FooocusExpansion:
def __call__(self, prompt, seed):
seed = int(seed)
set_seed(seed)
prompt = safe_str(prompt) + fooocus_magic_split[seed % len(fooocus_magic_split)]
origin = safe_str(prompt)
prompt = origin + fooocus_magic_split[seed % len(fooocus_magic_split)]
response = self.pipe(prompt, max_length=len(prompt) + 256)
result = response[0]['generated_text']
result = response[0]['generated_text'][len(origin):]
result = safe_str(result)
return result