mirrors/Fooocus
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix do_sample bug

Browse Source
This commit is contained in:
lvmin 2023-09-13 18:56:28 -07:00
parent 0f58038d85
commit db8748a34d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/expansion.py
View File

@ -54,7 +54,7 @@ class FooocusExpansion:
# https://huggingface.co/docs/transformers/generation_strategies # https://huggingface.co/docs/transformers/generation_strategies
features = self.model.generate(**tokenized_kwargs, features = self.model.generate(**tokenized_kwargs,
penalty_alpha=0.6, penalty_alpha=0.6,
top_k=4, top_k=8,
max_new_tokens=256, max_new_tokens=256,
do_sample=True) do_sample=True)