Update generate.py

generate.py

@@ -158,7 +158,7 @@ def generate(
 ):
     # Set default value for max_new_tokens if not provided
     if max_new_tokens is None:
-        max_new_tokens = 20  # Set a reasonable default value
+        max_new_tokens = 128  # Set a reasonable default value
 
     # Set model attributes
     self.max_thoughts = n_ahead + n_ahead_talk + 1