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

fix default for plugins

This commit is contained in:
Victor Hall 2023-06-29 22:00:16 -04:00
parent a72d455fc5
commit 1afaf59ec9
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
train.py
View File

@ -749,7 +749,10 @@ def main(args):
generate_samples(global_step=0, batch=batch) generate_samples(global_step=0, batch=batch)
from plugins.plugins import load_plugin from plugins.plugins import load_plugin
plugins = [load_plugin(name) for name in args.plugins] if args.plugins is not None:
plugins = [load_plugin(name) for name in args.plugins]
else:
plugins = []
try: try:
write_batch_schedule(args, log_folder, train_batch, epoch = 0) write_batch_schedule(args, log_folder, train_batch, epoch = 0)