From 74f54a5fef5952a0be143127676720d45d031d50 Mon Sep 17 00:00:00 2001 From: Max Woolf Date: Sun, 17 May 2020 12:43:18 -0700 Subject: [PATCH] Use 0.7 temperature default (matches gpt-2-simple) --- aitextgen/aitextgen.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aitextgen/aitextgen.py b/aitextgen/aitextgen.py index 7ca2e24..af956de 100644 --- a/aitextgen/aitextgen.py +++ b/aitextgen/aitextgen.py @@ -235,7 +235,7 @@ def generate( n: int = 1, prompt: str = None, max_length: int = 200, - temperature: float = 1.0, + temperature: float = 0.7, do_sample: bool = True, return_as_list: bool = False, seed: int = None,