Fix max_tokens (#1148)

2025-08-29 07:30:06 +08:00 · 2024-12-10 11:26:04 -08:00 · 2024-12-10 11:26:04 -08:00 · b83a73049d
commit b83a73049d
parent 64781fddf1
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,
        ):