From 1fa3a10e3d0373d4bf764ea4dc1f71836c080255 Mon Sep 17 00:00:00 2001 From: Angelos Katharopoulos Date: Sun, 8 Dec 2024 22:56:55 -0800 Subject: [PATCH] Revert "fix missing --final-adapter-file crash in dreambooth.py" This reverts commit f00ca80fdb154bb91339f5582dba600194d417ce. --- flux/dreambooth.py | 10 ++-------- 1 file changed, 2 insertions(+), 8 deletions(-) diff --git a/flux/dreambooth.py b/flux/dreambooth.py index 441659c2..ffdb02d7 100644 --- a/flux/dreambooth.py +++ b/flux/dreambooth.py @@ -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}.")