From 51658f50f0b44df5e923bdd35f706d9e3cbb2140 Mon Sep 17 00:00:00 2001 From: Turgut Date: Mon, 27 Apr 2026 09:54:49 +0200 Subject: [PATCH] fix --rng-deterministic argparse bool parsing --- codegen1/jaxformer/hf/sample.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codegen1/jaxformer/hf/sample.py b/codegen1/jaxformer/hf/sample.py index 4477e10b..755eb7b4 100644 --- a/codegen1/jaxformer/hf/sample.py +++ b/codegen1/jaxformer/hf/sample.py @@ -192,7 +192,7 @@ def main(): parser.add_argument('--model', type=str, choices=models, default='codegen-350M-mono') parser.add_argument('--device', type=str, default='cuda:0') parser.add_argument('--rng-seed', type=int, default=42) - parser.add_argument('--rng-deterministic', type=bool, default=True) + parser.add_argument('--rng-deterministic', action=argparse.BooleanOptionalAction, default=True) parser.add_argument('--p', type=float, default=0.95) parser.add_argument('--t', type=float, default=0.2) parser.add_argument('--max-length', type=int, default=128)