From ea2c0d233eb5dc485be905cdc2fc406658410cd6 Mon Sep 17 00:00:00 2001
From: Nick Walton <nickwalton00@gmail.com>
Date: Mon, 18 Nov 2019 18:48:17 -0700
Subject: [PATCH] update

---
 generator/gpt2/gpt2_generator.py | 2 +-
 generator/gpt2/src/sample.py     | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/generator/gpt2/gpt2_generator.py b/generator/gpt2/gpt2_generator.py
index 02df39a..234ba91 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=80, temperature=0.4, top_k=40, top_p=0.8):
+    def __init__(self,  generate_num=80, temperature=0.3, top_k=40, top_p=0.8):
         self.generate_num=generate_num
         self.temp = temperature
         self.top_k = top_k
diff --git a/generator/gpt2/src/sample.py b/generator/gpt2/src/sample.py
index 5879d11..220ff1d 100644
--- a/generator/gpt2/src/sample.py
+++ b/generator/gpt2/src/sample.py
@@ -17,7 +17,7 @@ def penalize_used(logits, output):
 
     return tf.compat.v1.where(
             bool_tensor,
-            logits / 1.2,
+            logits / 1.1,
             logits)