diff --git a/generator/gpt2/gpt2_generator.py b/generator/gpt2/gpt2_generator.py index 66a0813..f64deef 100644 --- a/generator/gpt2/gpt2_generator.py +++ b/generator/gpt2/gpt2_generator.py @@ -10,7 +10,7 @@ import numpy as np class GPT2Generator: - def __init__(self, generate_num=60, temperature=0.3, top_k=40, top_p=0.9): + def __init__(self, generate_num=60, temperature=0.4, top_k=40, top_p=0.9): self.generate_num=generate_num self.temp = temperature self.top_k = top_k diff --git a/generator/gpt2/src/sample.py b/generator/gpt2/src/sample.py index 5879d11..f50eaf8 100644 --- a/generator/gpt2/src/sample.py +++ b/generator/gpt2/src/sample.py @@ -17,7 +17,7 @@ def penalize_used(logits, output): return tf.compat.v1.where( bool_tensor, - logits / 1.2, + logits * 0.85, logits) diff --git a/play.py b/play.py index 529b169..20dd1a1 100644 --- a/play.py +++ b/play.py @@ -34,7 +34,7 @@ def play_aidungeon_2(): save_story = input("Help improve AIDungeon by enabling story saving? (Y/n) ") if save_story.lower() in ["no", "No", "n"]: - upload_story = True + upload_story = False else: upload_story = True