Fix max_tokens (#1148)

2025-09-01 04:14:38 +08:00 · 2024-12-10 11:26:04 -08:00
parent 12083c4b7e
commit 135c5818c1
1 changed files with 1 additions and 1 deletions
--- a/llms/mlx_lm/chat.py
+++ b/llms/mlx_lm/chat.py
@@ -79,7 +79,7 @@ def main():
            model,
            tokenizer,
            prompt,
-            args.max_tokens,
+            max_tokens=args.max_tokens,
            sampler=make_sampler(args.temp, args.top_p),
            prompt_cache=prompt_cache,
        ):