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

Added Model rescale and prepared a release upgrade

Browse Source
This commit is contained in:
Jaret Burkett
2023-08-01 13:49:54 -06:00
parent 63cacf4362
commit 8b8d53888d
15 changed files with 388 additions and 64 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
toolkit/job.py
View File

@@ -13,6 +13,9 @@ def get_job(config_path, name=None):
if job == 'train':
from jobs import TrainJob
return TrainJob(config)
if job == 'mod':
from jobs import ModJob
return ModJob(config)
# elif job == 'train':
# from jobs import TrainJob