ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-01-26 16:39:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Send more data when loading the model to the ui

Browse Source
This commit is contained in:
Jaret Burkett
2025-02-23 12:49:54 -07:00
parent b366e46f1c
commit 60f848a877
3 changed files with 45 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
jobs/process/BaseSDTrainProcess.py
View File

@@ -730,6 +730,9 @@ class BaseSDTrainProcess(BaseTrainProcess):
def hook_train_loop(self, batch):
# return loss
return 0.0
def hook_after_sd_init_before_load(self):
pass
def get_latest_save_path(self, name=None, post=''):
if name == None:
@@ -1425,6 +1428,8 @@ class BaseSDTrainProcess(BaseTrainProcess):
custom_pipeline=self.custom_pipeline,
noise_scheduler=sampler,
)
self.hook_after_sd_init_before_load()
# run base sd process run
self.sd.load_model()