From 6cca16f97a5a3c1844898567d95a6df15584e442 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Mon, 1 Jun 2026 14:43:48 +0000 Subject: [PATCH] Set max-tokens=128 default, clean up for final verification --- 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 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)