From a0e758ec3b6310ab12313a14c6cef06e2757b754 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Mon, 1 Jun 2026 14:33:55 +0000 Subject: [PATCH] Set default max-tokens=30 for faster iteration --- single_shot_inference.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/single_shot_inference.py b/single_shot_inference.py index 88e3973c..da4ed0f9 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=8192) + p.add_argument('--max-tokens', type=int, default=30) p.add_argument('--prompt', type=str, default=None) p.add_argument('--seed', type=int, default=42) p.add_argument('--verbose', type=int, default=1)