ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-05-01 03:31:35 +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

9
extensions_built_in/sd_trainer/UITrainer.py
View File

@@ -155,6 +155,15 @@ class UITrainer(SDTrainer):
super().hook_before_train_loop()
self.maybe_stop()
self.update_status("running", "Training")
def status_update_hook_func(self, string):
self.update_status("running", string)
def hook_after_sd_init_before_load(self):
super().hook_after_sd_init_before_load()
self.maybe_stop()
self.sd.add_status_update_hook(self.status_update_hook_func)
def sample_step_hook(self, img_num, total_imgs):
super().sample_step_hook(img_num, total_imgs)