condition and supervise

lucidtronix · lucidtronix · commit 21039b185e3e · 2025-01-17T13:47:00.000-05:00
diff --git a/ml4h/models/diffusion_blocks.py b/ml4h/models/diffusion_blocks.py
@@ -877,7 +877,7 @@ def test_step(self, batch):
         # this is computationally demanding, kid_diffusion_steps has to be small
         if self.input_map.axes() == 3 and self.inspect_model:
             images = self.denormalize(images)
-            generated_images = self.generate(
+            generated_images = self.generate(control_embed,
                 num_images=self.batch_size, diffusion_steps=20
             )
             self.kid.update_state(images, generated_images)

Original file line number	Diff line number	Diff line change
`@@ -877,7 +877,7 @@ def test_step(self, batch):`
`877`	`877`	`# this is computationally demanding, kid_diffusion_steps has to be small`
`878`	`878`	`if self.input_map.axes() == 3 and self.inspect_model:`
`879`	`879`	`images = self.denormalize(images)`
`880`		`- generated_images = self.generate(`
	`880`	`+ generated_images = self.generate(control_embed,`
`881`	`881`	`num_images=self.batch_size, diffusion_steps=20`
`882`	`882`	`)`
`883`	`883`	`self.kid.update_state(images, generated_images)`