Big refactor of SD runner and added image generator

2023-08-03 14:51:25 -06:00
parent 75ec5d9292
commit 66c6f0f6f7
16 changed files with 923 additions and 430 deletions
--- a/toolkit/job.py
+++ b/toolkit/job.py
@@ -16,6 +16,9 @@ def get_job(config_path, name=None):
    if job == 'mod':
        from jobs import ModJob
        return ModJob(config)
+    if job == 'generate':
+        from jobs import GenerateJob
+        return GenerateJob(config)

    # elif job == 'train':
    #     from jobs import TrainJob