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

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

Browse Source 
This reverts commit f00ca80fdb.
This commit is contained in:
Angelos Katharopoulos 2024-12-08 22:56:55 -08:00
parent f00ca80fdb
commit 1fa3a10e3d
1 changed files with 2 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
flux/dreambooth.py
View File

@ -149,12 +149,6 @@ 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
@ -294,5 +288,5 @@ if __name__ == "__main__":
losses = []
tic = time.time()
save_adapters(args.final_adapter_file, flux, args)
print(f"Training successful. Saved final weights to {args.final_adapter_file}.")
save_adapters("final_adapters.safetensors", flux, args)
print(f"Training successful. Saved final weights to {args.adapter_file}.")