missed issue with plugin

2023-06-29 20:44:14 -04:00 · 2023-06-29 20:44:14 -04:00 · a72d455fc5
parent e00b7b571a
commit a72d455fc5
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -748,7 +748,7 @@ def main(args):
        _, batch = next(enumerate(train_dataloader))
        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]

    try: