diff --git a/generator/gpt2/gpt2_generator.py b/generator/gpt2/gpt2_generator.py index 1a07568..202f24f 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.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 diff --git a/play.py b/play.py index b7015a9..e1224f3 100644 --- a/play.py +++ b/play.py @@ -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("> ")