set beam to 1 for balanced distrubution

2023-09-13 19:26:34 -07:00 · 2023-09-13 19:26:34 -07:00 · 917f284db1
parent db8748a34d
commit 917f284db1
1 changed files with 1 additions and 2 deletions
--- 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)