Do a noisy unconsitional for vision direct

2024-07-29 15:42:26 -06:00 · 2024-07-29 15:42:26 -06:00 · 443c996e7f
parent 8f0f467c20
commit 443c996e7f
1 changed files with 1 additions and 1 deletions
--- a/extensions_built_in/sd_trainer/SDTrainer.py
+++ b/extensions_built_in/sd_trainer/SDTrainer.py
@ -1174,7 +1174,7 @@ class SDTrainer(BaseSDTrainProcess):
                    quad_count = random.randint(1, 4)
                    self.adapter.train()
                    self.adapter.trigger_pre_te(
-                        tensors_0_1=clip_images,
+                        tensors_0_1=clip_images if not is_reg else None,  # on regs we send none to get random noise
                        is_training=True,
                        has_been_preprocessed=True,
                        quad_count=quad_count,