Fix Qwen3 thinking mode + increase max_new_tokens: training/eval_adapter.py
Browse files- 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=
|
| 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()
|