bug fixes

2023-09-09 15:04:44 -06:00
parent 2128ac1e08
commit 626ed2939a
2 changed files with 10 additions and 10 deletions
--- a/jobs/process/BaseSDTrainProcess.py
+++ b/jobs/process/BaseSDTrainProcess.py
@@ -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()