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

This commit is contained in:
Jaret Burkett committed 2023-07-08 13:50:59 -06:00
1 parent 37354b006e
commit 47d094e528
13 files changed
+151 -18

No files matched your search

+3
View File
@@ -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}')