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

Fixing "NameError: name 'resume_adapter_file' is not defined" (#817)

Browse Source 
args. is missing from resume_adapter_file so the name is not defined.
This commit is contained in:
Robin Glauser 2024-06-05 19:07:31 +02:00 committed by GitHub
parent 43d6deb3c1
commit 4872727f14
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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

@ -170,7 +170,7 @@ def train_model(
# Resume training the given adapters. # Resume training the given adapters.
if args.resume_adapter_file is not None: if args.resume_adapter_file is not None:
print(f"Loading pretrained adapters from {resume_adapter_file}") print(f"Loading pretrained adapters from {args.resume_adapter_file}")
model.load_weights(args.resume_adapter_file, strict=False) model.load_weights(args.resume_adapter_file, strict=False)
print_trainable_parameters(model) print_trainable_parameters(model)