From 0d42c42bfbf9e895cafce6faad4fc6bb46e1fb63 Mon Sep 17 00:00:00 2001 From: Nick Walton Date: Mon, 18 Nov 2019 23:10:40 -0700 Subject: [PATCH] update --- generator/gpt2/gpt2_generator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/generator/gpt2/gpt2_generator.py b/generator/gpt2/gpt2_generator.py index ab2bb11..3b9ce22 100644 --- a/generator/gpt2/gpt2_generator.py +++ b/generator/gpt2/gpt2_generator.py @@ -11,7 +11,7 @@ tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.ERROR) class GPT2Generator: - def __init__(self, generate_num=64, temperature=0.4, top_k=40, top_p=0.9): + def __init__(self, generate_num=56, temperature=0.4, top_k=40, top_p=0.9): self.generate_num=generate_num self.temp = temperature self.top_k = top_k