mirror of
https://github.com/ostris/ai-toolkit.git
synced 2026-04-29 10:41:28 +00:00
Setup base for training jobs. Added sd-scripts as a submodule
This commit is contained in:
25
jobs/process/BaseTrainProcess.py
Normal file
25
jobs/process/BaseTrainProcess.py
Normal file
@@ -0,0 +1,25 @@
|
||||
from collections import OrderedDict
|
||||
from jobs import TrainJob
|
||||
from jobs.process.BaseProcess import BaseProcess
|
||||
|
||||
|
||||
class BaseTrainProcess(BaseProcess):
|
||||
job: TrainJob
|
||||
process_id: int
|
||||
config: OrderedDict
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
process_id: int,
|
||||
job: TrainJob,
|
||||
config: OrderedDict
|
||||
):
|
||||
super().__init__(process_id, job, config)
|
||||
self.process_id = process_id
|
||||
self.job = job
|
||||
self.config = config
|
||||
|
||||
def run(self):
|
||||
# implement in child class
|
||||
# be sure to call super().run() first
|
||||
pass
|
||||
Reference in New Issue
Block a user