Update generate.py

generate.py

@@ -40,6 +40,8 @@ def custom_generate(
     forced_bos_token_id=None,
     forced_eos_token_id=None,
     remove_invalid_values=None,
+	n_ahead=12,
+    n_ahead_talk=4,
     synced_gpus=None,
     **kwargs,
 ):