mirror of
https://github.com/lllyasviel/Fooocus.git
synced 2026-08-16 13:13:16 +02:00
fix do_sample bug
This commit is contained in:
+1
-1
@@ -1 +1 @@
|
||||
version = '2.0.10'
|
||||
version = '2.0.11'
|
||||
|
||||
@@ -52,7 +52,11 @@ class FooocusExpansion:
|
||||
|
||||
# https://huggingface.co/blog/introducing-csearch
|
||||
# https://huggingface.co/docs/transformers/generation_strategies
|
||||
features = self.model.generate(**tokenized_kwargs, penalty_alpha=0.8, top_k=8, max_new_tokens=256)
|
||||
features = self.model.generate(**tokenized_kwargs,
|
||||
penalty_alpha=0.6,
|
||||
top_k=4,
|
||||
max_new_tokens=256,
|
||||
do_sample=True)
|
||||
|
||||
response = self.tokenizer.batch_decode(features, skip_special_tokens=True)
|
||||
result = response[0][len(origin):]
|
||||
|
||||
Reference in New Issue
Block a user