From 305e898c56d7fd560b94444f85f15b3b7d07900e Mon Sep 17 00:00:00 2001
From: Nick Walton <nickwalton00@gmail.com>
Date: Mon, 18 Nov 2019 12:40:49 -0700
Subject: [PATCH] added print wrapping

---
 generator/gpt2/gpt2_generator.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/generator/gpt2/gpt2_generator.py b/generator/gpt2/gpt2_generator.py
index 947ad27..9a87608 100644
--- a/generator/gpt2/gpt2_generator.py
+++ b/generator/gpt2/gpt2_generator.py
@@ -15,7 +15,7 @@ tf.logging.set_verbosity(tf.logging.ERROR)
 
 class GPT2Generator:
 
-    def __init__(self,  generate_num=80, temperature=0.3, top_k=40, top_p=0.8):
+    def __init__(self,  generate_num=100, temperature=0.3, top_k=40, top_p=0.8):
         self.generate_num=generate_num
         self.temp = temperature
         self.top_k = top_k
@@ -90,7 +90,7 @@ class GPT2Generator:
 
     def generate(self, prompt, options=None, seed=1):
 
-        debug_print=True
+        debug_print=False
         prefix = self.prompt_replace(prompt)
 
         if debug_print: