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

missed issue with plugin

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

2
train.py
View File

@ -748,7 +748,7 @@ def main(args):
_, batch = next(enumerate(train_dataloader)) _, batch = next(enumerate(train_dataloader))
generate_samples(global_step=0, batch=batch) generate_samples(global_step=0, batch=batch)
from plugins.base_plugin import load_plugin from plugins.plugins import load_plugin
plugins = [load_plugin(name) for name in args.plugins] plugins = [load_plugin(name) for name in args.plugins]
try: try: