From 917f284db1883cd9aabc79985880f5c589a402b7 Mon Sep 17 00:00:00 2001
From: lvmin <lyuminzhang@outlook.com>
Date: Wed, 13 Sep 2023 19:26:34 -0700
Subject: [PATCH] set beam to 1 for balanced distrubution

---
 modules/expansion.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/modules/expansion.py b/modules/expansion.py
index dd3d35e4..83d38906 100644
--- a/modules/expansion.py
+++ b/modules/expansion.py
@@ -53,8 +53,7 @@ class FooocusExpansion:
         # https://huggingface.co/blog/introducing-csearch
         # https://huggingface.co/docs/transformers/generation_strategies
         features = self.model.generate(**tokenized_kwargs,
-                                       penalty_alpha=0.6,
-                                       top_k=8,
+                                       num_beams=1,
                                        max_new_tokens=256,
                                        do_sample=True)