happyz
/
Fooocus
mirror of https://github.com/lllyasviel/Fooocus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

maintain gpt 

maintain gpt
This commit is contained in:
lllyasviel 2023-10-31 12:32:02 -07:00
parent 1b96d3ba0b
commit f30e71ad64
3 changed files with 996 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
fooocus_version.py
View File

@ -1 +1 @@
version = '2.1.769'
version = '2.1.770'

992
models/prompt_expansion/fooocus_expansion/positive.txt
View File

File diff suppressed because one or more lines are too long

8
modules/expansion.py
View File

@ -29,14 +29,14 @@ class FooocusExpansion:
def __init__(self):
self.tokenizer = AutoTokenizer.from_pretrained(fooocus_expansion_path)
positive_words = open(os.path.join(fooocus_expansion_path, 'positive.txt'), encoding='utf-8').read()
positive_words = positive_words.lower().replace(' ', '').replace('\n', '').split(',')
# print(', '.join(sorted(list(set(positive_words)))))
positive_words = open(os.path.join(fooocus_expansion_path, 'positive.txt'),
encoding='utf-8').read().splitlines()
# new_content = '\n'.join(sorted(list(set(positive_words))))
# t198 = self.tokenizer('\n', return_tensors="np")
# t11 = self.tokenizer(',', return_tensors="np")
# positive_ids = [11, 198, self.tokenizer.eos_token_id]
positive_ids = [11]
self.bad_words_ids = []