Hotfix for top_p parameter in generator call

2021-05-11 14:16:34 -04:00 · 2021-05-11 14:16:34 -04:00 · fff77c5a88
parent f520e859cf
commit fff77c5a88
1 changed files with 1 additions and 0 deletions
--- a/aiserver.py
+++ b/aiserver.py
@ -642,6 +642,7 @@ def generate(txt, min, max):
        min_length=min, 
        max_length=max,
        repetition_penalty=vars.rep_pen,
+        top_p=vars.top_p,
        temperature=vars.temp
        )[0]["generated_text"]
    print("{0}{1}{2}".format(colors.CYAN, genout, colors.END))