From d765aa5c9d555ca1e401d101d92d2967c98720eb Mon Sep 17 00:00:00 2001 From: Max Woolf Date: Sun, 17 May 2020 14:09:49 -0700 Subject: [PATCH] set default back to 256 for user-friendlyness --- aitextgen/aitextgen.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aitextgen/aitextgen.py b/aitextgen/aitextgen.py index 4e76eb7..5c2a6a6 100644 --- 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,