From 09f5add15149ccc27a67c680605da3113a4db0d7 Mon Sep 17 00:00:00 2001 From: paNikitin <115797306+paNikitin@users.noreply.github.com> Date: Mon, 24 Feb 2025 19:30:17 +0300 Subject: [PATCH] Update lora.py --- llms/mlx_lm/lora.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llms/mlx_lm/lora.py b/llms/mlx_lm/lora.py index eb0a279e..5f24da61 100644 --- a/llms/mlx_lm/lora.py +++ b/llms/mlx_lm/lora.py @@ -215,7 +215,7 @@ def train_model( adapter_path.mkdir(parents=True, exist_ok=True) adapter_file = adapter_path / "adapters.safetensors" - save_config(vars(args), adapter_path / "adapter_config.json") + save_config(vars(args), tokenizer, adapter_path / "adapter_config.json") # init training args training_args = TrainingArgs(