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

fix missing --final-adapter-file crash in dreambooth.py

Browse Source
This commit is contained in:
Peter Sibley 2024-12-06 22:23:55 -05:00
parent 1727959a27
commit f00ca80fdb
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
flux/dreambooth.py
View File

@ -149,6 +149,12 @@ def setup_arg_parser():
"--output-dir", default="mlx_output", help="Folder to save the checkpoints in"
)
parser.add_argument(
"--final-adapter-file",
default="final_adapters.safetensors",
help="The file name in the output-dir of the final adapter including the extension .safetensors",
)
parser.add_argument("dataset")
return parser
@ -288,5 +294,5 @@ if __name__ == "__main__":
losses = []
tic = time.time()
save_adapters("final_adapters.safetensors", flux, args)
print(f"Training successful. Saved final weights to {args.adapter_file}.")
save_adapters(args.final_adapter_file, flux, args)
print(f"Training successful. Saved final weights to {args.final_adapter_file}.")