Mirrors
/
EveryDream2trainer
mirror of https://github.com/victorchall/EveryDream2trainer.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

bugged flag

This commit is contained in:
Victor Hall 2023-09-22 10:16:56 -04:00
parent 166c2e74e1
commit a9c98f5866
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
train.py
View File

@ -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,