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

bug fixes

Browse Source
This commit is contained in:
Jaret Burkett committed 2023-09-09 15:04:44 -06:00
1 parent 2128ac1e08
commit 626ed2939a
2 files changed
+10 -10
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

jobs/process/BaseSDTrainProcess.py
+1 -1
View File
@@ -733,7 +733,7 @@ class BaseSDTrainProcess(BaseTrainProcess):
else:
self.sd.text_encoder.train()
self.sd.unet.train()
self.sd.unet.train()
### HOOK ###
loss_dict = self.hook_train_loop(batch)
flush()