diff --git a/console_play.py b/console_play.py index 48f809a..333633c 100644 --- a/console_play.py +++ b/console_play.py @@ -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 diff --git a/generator/gpt2/gpt2_generator.py b/generator/gpt2/gpt2_generator.py index 3b9ce22..5ac4497 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=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