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,