Added extensions and an example extension that merges models

This commit is contained in:
Jaret Burkett
2023-08-04 09:37:24 -06:00
parent b865ac8b24
commit 7e4e660663
14 changed files with 366 additions and 24 deletions

View File

@@ -19,6 +19,9 @@ def get_job(config_path, name=None):
if job == 'generate':
from jobs import GenerateJob
return GenerateJob(config)
if job == 'extension':
from jobs import ExtensionJob
return ExtensionJob(config)
# elif job == 'train':
# from jobs import TrainJob