Set max-tokens=128 default, clean up for final verification

2026-06-01 14:43:48 +00:00
parent a0e758ec3b
commit 6cca16f97a
1 changed files with 1 additions and 1 deletions
--- 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)