zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-08-29 09:56:24 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update lora.py

Browse Source
This commit is contained in:
paNikitin 2025-02-24 19:30:17 +03:00
parent b2ab37238e
commit 09f5add151
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
llms/mlx_lm/lora.py
View File

@ -215,7 +215,7 @@ def train_model(
adapter_path.mkdir(parents=True, exist_ok=True) adapter_path.mkdir(parents=True, exist_ok=True)
adapter_file = adapter_path / "adapters.safetensors" 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 # init training args
training_args = TrainingArgs( training_args = TrainingArgs(