Increase the temperature to 0.9.

1 files changed, 1 insertions, 1 deletions

diff --git a/ai_dungeon.py b/ai_dungeon.py
index eb44a87..9d28bb3 100755
--- a/ai_dungeon.py
+++ b/ai_dungeon.py
@@ -29,7 +29,7 @@ def trim_sentence(message: str) -> str:
 
 def generate(model: transformers.TextGenerationPipeline, prompt: str, **kwargs) -> str:
     """Generate text from a text generator."""
-    out = model(prompt, do_sample=True, temperature=0.8, top_k=60, top_p=0.9, **kwargs)
+    out = model(prompt, do_sample=True, temperature=0.9, top_k=60, top_p=0.9, **kwargs)
     return trim_sentence(out[0]["generated_text"])