WIP implementing training

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

View File

@@ -1,19 +1,16 @@
from toolkit.kohya_model_util import load_models_from_stable_diffusion_checkpoint
from .BaseJob import BaseJob
from collections import OrderedDict
from typing import List
from jobs.process import BaseExtractProcess
from jobs.process import ExtractLoconProcess
from jobs import BaseJob
process_dict = {
'locon': ExtractLoconProcess,
'locon': 'ExtractLoconProcess',
}
class ExtractJob(BaseJob):
process: List[BaseExtractProcess]
def __init__(self, config: OrderedDict):
super().__init__(config)