set default back to 256 for user-friendlyness

minimaxir · May 17, 2020 · d765aa5 · d765aa5
1 parent 1df1211
commit d765aa5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/aitextgen/aitextgen.py b/aitextgen/aitextgen.py
@@ -234,7 +234,7 @@ def generate(
         self,
         n: int = 1,
         prompt: str = None,
-        max_length: int = 1024,
+        max_length: int = 256,
         temperature: float = 0.7,
         do_sample: bool = True,
         return_as_list: bool = False,