Use 0.7 temperature default (matches gpt-2-simple)

minimaxir · May 17, 2020 · 74f54a5 · 74f54a5
1 parent 4015537
commit 74f54a5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,