import gcd

app.py

@@ -5,6 +5,7 @@ from gradio import update
 from functools import lru_cache
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 from opencc import OpenCC  # 用於簡體轉繁體
+from math import gcd
 
 # 初始化簡體到繁體轉換器
 cc = OpenCC('s2t')
@@ -81,7 +82,7 @@ def suggest_next(text, model_name, k, m, num_beam_groups, diversity_penalty):
     }
     if diversity_penalty and diversity_penalty > 0:
         valid_group = gcd(m, num_beam_groups)
-        gen_kwargs["num_beam_groups"] = num_beam_groups
+        gen_kwargs["num_beam_groups"] = valid_group
         gen_kwargs["diversity_penalty"] = float(diversity_penalty)
 
     outs = gen_pipe(text, **gen_kwargs)