piyush-mk commited on
Commit
664fa8e
·
verified ·
1 Parent(s): 7c48dcf

Fix Qwen3 thinking mode + increase max_new_tokens: training/eval_adapter.py

Browse files
Files changed (1) hide show
  1. training/eval_adapter.py +1 -1
training/eval_adapter.py CHANGED
@@ -79,7 +79,7 @@ def main() -> None:
79
  p.add_argument("--adapter-repo", required=True)
80
  p.add_argument("--slice", choices=["canonical", "hard", "all"], default="all")
81
  p.add_argument("--max-tasks", type=int, default=None)
82
- p.add_argument("--max-new-tokens", type=int, default=96)
83
  p.add_argument("--max-prompt-tokens", type=int, default=2048)
84
  p.add_argument("--artifact-dir", default="/tmp/invoiceguard-adapter-eval")
85
  args = p.parse_args()
 
79
  p.add_argument("--adapter-repo", required=True)
80
  p.add_argument("--slice", choices=["canonical", "hard", "all"], default="all")
81
  p.add_argument("--max-tasks", type=int, default=None)
82
+ p.add_argument("--max-new-tokens", type=int, default=384)
83
  p.add_argument("--max-prompt-tokens", type=int, default=2048)
84
  p.add_argument("--artifact-dir", default="/tmp/invoiceguard-adapter-eval")
85
  args = p.parse_args()