From 443c996e7fff55219e130c8e14c4836ce3ac50bf Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Mon, 29 Jul 2024 15:42:26 -0600 Subject: [PATCH] Do a noisy unconsitional for vision direct --- extensions_built_in/sd_trainer/SDTrainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/extensions_built_in/sd_trainer/SDTrainer.py b/extensions_built_in/sd_trainer/SDTrainer.py index e4468575..e1450593 100644 --- 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,