Added extensions and an example extension that merges models

This commit is contained in:
Jaret Burkett committed 2023-08-04 09:37:24 -06:00
1 parent b865ac8b24
commit 7e4e660663
14 files changed
+366 -24

No files matched your search

+3
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