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

fix loading adapters

Browse Source
This commit is contained in:
Awni Hannun
2024-01-04 11:54:49 -08:00
parent 0c140c9019
commit 8b960fc3e6
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lora/lora.py
View File

@@ -394,7 +394,7 @@ if __name__ == "__main__":
f"Adapter file {args.adapter_file} missing. " f"Adapter file {args.adapter_file} missing. "
"Use --train to learn and save the adapters.npz." "Use --train to learn and save the adapters.npz."
) )
model.load_weights(args.adapter_file) model.load_weights(args.adapter_file, strict=False)
if args.test: if args.test:
print("Testing") print("Testing")