Setup base for training jobs. Added sd-scripts as a submodule
This commit is contained in:
1 parent
37354b006e
commit
47d094e528
13 files changed
+151
-18
No files matched your search
@@ -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}')
|
||||
Reference in new issue
Block a user