diff options
Diffstat (limited to 'ai_dungeon.py')
| -rwxr-xr-x | ai_dungeon.py | 2 |
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"]) |