This commit is contained in:
Nick Walton
2019-11-18 22:47:40 -07:00
parent 207b3b7b11
commit bdd092e641
2 changed files with 2 additions and 2 deletions
+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=80, temperature=0.3, top_k=40, top_p=0.8):
def __init__(self, generate_num=64, temperature=0.3, top_k=40, top_p=0.8):
self.generate_num=generate_num
self.temp = temperature
self.top_k = top_k
+1 -1
View File
@@ -58,7 +58,7 @@ def play_aidungeon_2():
print(starter)
print("\n")
console_print(context + str(story_manager.story))
console_print(context + "\n\n" + str(story_manager.story))
while True:
action = input("> ")