github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Added inbuild plugins and made one for image referenced. WIP

Browse Source
This commit is contained in:
Jaret Burkett
2023-08-10 16:02:44 -06:00
parent df48f0a843
commit 1a7e346b41
12 changed files with 338 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
jobs/process/TrainLoRAHack.py
View File

@@ -68,7 +68,7 @@ class TrainLoRAHack(BaseSDTrainProcess):
return loss_dict
def hook_train_loop(self):
def hook_train_loop(self, batch):
if self.hack_config.type == 'suppression':
return self.supress_loop()
else: