This commit is contained in:
Nick Walton committed 2019-11-18 23:15:48 -07:00
1 parent ad2ab0aede
commit 9418fe6d5f
2 files changed
+1 -3

No files matched your search

-2
View File
@@ -1,8 +1,6 @@
from story.story_manager import *
from generator.gpt2.gpt2_generator import *
def console_print(text, width=75):
last_newline = 0
i = 0
+1 -1
View File
@@ -11,7 +11,7 @@ tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.ERROR)
class GPT2Generator:
def __init__(self, generate_num=56, temperature=0.4, top_k=40, top_p=0.9):
def __init__(self, generate_num=48, temperature=0.4, top_k=40, top_p=0.9):
self.generate_num=generate_num
self.temp = temperature
self.top_k = top_k