ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-04-30 03:01:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

WIP implementing training

Browse Source
This commit is contained in:
Jaret Burkett
2023-07-12 08:23:46 -06:00
parent 47d094e528
commit 57f14e5ef2
16 changed files with 1031 additions and 67 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
toolkit/job.py
View File

@@ -1,8 +1,7 @@
from jobs import BaseJob
from toolkit.config import get_config
def get_job(config_path) -> BaseJob:
def get_job(config_path):
config = get_config(config_path)
if not config['job']:
raise ValueError('config file is invalid. Missing "job" key')
@@ -11,8 +10,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)
# elif job == 'train':
# from jobs import TrainJob
# return TrainJob(config)
else:
raise ValueError(f'Unknown job type {job}')