diff --git a/single_shot_inference.py b/single_shot_inference.py
index da4ed0f9..7ed72262 100644
--- a/single_shot_inference.py
+++ b/single_shot_inference.py
@@ -18,7 +18,7 @@ log = logging.getLogger("single_shot")
 
 def parse_args():
     p = argparse.ArgumentParser()
-    p.add_argument('--max-tokens', type=int, default=30)
+    p.add_argument('--max-tokens', type=int, default=128)
     p.add_argument('--prompt', type=str, default=None)
     p.add_argument('--seed', type=int, default=42)
     p.add_argument('--verbose', type=int, default=1)