From 0f58038d856682356ef51209b164261b14732898 Mon Sep 17 00:00:00 2001 From: lvmin Date: Wed, 13 Sep 2023 18:55:51 -0700 Subject: [PATCH] fix do_sample bug --- fooocus_version.py | 2 +- modules/expansion.py | 6 +++++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/fooocus_version.py b/fooocus_version.py index 03f874b9..054f64ca 100644 --- a/fooocus_version.py +++ b/fooocus_version.py @@ -1 +1 @@ -version = '2.0.10' +version = '2.0.11' diff --git a/modules/expansion.py b/modules/expansion.py index 3dc1b8b8..5059fdb8 100644 --- a/modules/expansion.py +++ b/modules/expansion.py @@ -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):]