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

1. Fix to EMA samples arguments not respecting False value.

This commit is contained in:
alexds9 2023-09-06 23:04:12 +03:00
parent 5bcf9407f0
commit cf4a082e11
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
train.py
View File

@ -917,10 +917,10 @@ def main(args):
models_info = []
if (args.ema_decay_rate is None) or (args.ema_decay_sample_raw_training is not None):
if (args.ema_decay_rate is None) or args.ema_decay_sample_raw_training:
models_info.append({"is_ema": False, "swap_required": False})
if (args.ema_decay_rate is not None) and (args.ema_decay_sample_ema_model is not None):
if (args.ema_decay_rate is not None) and args.ema_decay_sample_ema_model:
models_info.append({"is_ema": True, "swap_required": args.ema_decay_device != device})
for model_info in models_info: