From 739a3e62a05c8701eb1ac02f3e2e520d068435ab Mon Sep 17 00:00:00 2001 From: Anchen Date: Mon, 4 Nov 2024 14:58:17 +0800 Subject: [PATCH] chore: update the default max token value --- llms/mlx_lm/chat.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llms/mlx_lm/chat.py b/llms/mlx_lm/chat.py index 18ae38b2..85d32d5f 100644 --- a/llms/mlx_lm/chat.py +++ b/llms/mlx_lm/chat.py @@ -11,7 +11,7 @@ from .utils import load, stream_generate DEFAULT_TEMP = 0.0 DEFAULT_TOP_P = 1.0 DEFAULT_SEED = 0 -DEFAULT_MAX_TOKENS = 100 +DEFAULT_MAX_TOKENS = 256 DEFAULT_MODEL = "mlx-community/Llama-3.2-3B-Instruct-4bit"