From a9c98f586617489fd85082f0868cd8b89d818153 Mon Sep 17 00:00:00 2001 From: Victor Hall Date: Fri, 22 Sep 2023 10:16:56 -0400 Subject: [PATCH] bugged flag --- train.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/train.py b/train.py index 5e4f67b..1c6b220 100644 --- a/train.py +++ b/train.py @@ -159,7 +159,6 @@ def save_model(save_path, ed_state: EveryDreamTrainingState, global_step: int, s logging.warning(" No model to save, something likely blew up on startup, not saving") return - if args.ema_decay_rate != None: pipeline_ema = StableDiffusionPipeline( vae=ed_state.vae, @@ -1208,7 +1207,7 @@ def main(args): save_model(save_path, global_step=global_step, ed_state=make_current_ed_state(), save_ckpt_dir=None, yaml_name=None, save_full_precision=args.save_full_precision, - save_optimizer_flag=args.save_optimizer, save_ckpt=False) + save_optimizer_flag=args.save_optimizer, save_ckpt=not args.no_save_ckpt) plugin_runner.run_on_step_end(epoch=epoch, global_step=global_step,