mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-08-29 04:25:06 +08:00
Update lora.py
This commit is contained in:
parent
b2ab37238e
commit
09f5add151
@ -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(
|
||||
|
Loading…
Reference in New Issue
Block a user