Added extensions and an example extension that merges models

2023-08-04 09:37:24 -06:00
parent b865ac8b24
commit 7e4e660663
14 changed files with 366 additions and 24 deletions
--- a/toolkit/job.py
+++ b/toolkit/job.py
@@ -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