From 4817605a0c52a52dc1b9fcae6774887990eb8a3f Mon Sep 17 00:00:00 2001 From: Rose Hogenson Date: Tue, 15 Feb 2022 09:02:16 -0800 Subject: Increase the temperature to 0.9. --- ai_dungeon.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'ai_dungeon.py') 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"]) -- cgit v1.3.1