Setup base for training jobs. Added sd-scripts as a submodule

2026-04-29 02:31:17 +00:00 · 2023-07-08 13:50:59 -06:00
parent 37354b006e
commit 47d094e528
13 changed files with 151 additions and 18 deletions
--- a/toolkit/job.py
+++ b/toolkit/job.py
@@ -11,5 +11,8 @@ def get_job(config_path) -> BaseJob:
    if job == 'extract':
        from jobs import ExtractJob
        return ExtractJob(config)
+    elif job == 'train':
+        from jobs import TrainJob
+        return TrainJob(config)
    else:
        raise ValueError(f'Unknown job type {job}')