Added inbuild plugins and made one for image referenced. WIP

2023-08-10 16:02:44 -06:00
parent df48f0a843
commit 1a7e346b41
12 changed files with 338 additions and 26 deletions
--- a/jobs/process/TrainSliderProcess.py
+++ b/jobs/process/TrainSliderProcess.py
@@ -173,7 +173,7 @@ class TrainSliderProcess(BaseSDTrainProcess):
        flush()
        # end hook_before_train_loop

-    def hook_train_loop(self):
+    def hook_train_loop(self, batch):
        dtype = get_torch_dtype(self.train_config.dtype)

        # get a random pair