diff --git a/modules/expansion.py b/modules/expansion.py
index 5059fdb..dd3d35e 100644
--- a/modules/expansion.py
+++ b/modules/expansion.py
@@ -54,7 +54,7 @@ class FooocusExpansion:
         # https://huggingface.co/docs/transformers/generation_strategies
         features = self.model.generate(**tokenized_kwargs,
                                        penalty_alpha=0.6,
-                                       top_k=4,
+                                       top_k=8,
                                        max_new_tokens=256,
                                        do_sample=True)