Update main.py

main.py

@@ -70,8 +70,8 @@ class Upscale_CaseCade:
             cnet = self.models.controlnet(effnet_latents_up)
             cnet_uncond = cnet
             cnet_input = torch.nn.functional.interpolate(images, scale_factor=scale_fator, mode="nearest")
-            # cnet, cnet_input = self.core.get_cnet(batch, self.models, self.extras)
-            # cnet_uncond = cnet
+            cnet, cnet_input = self.core.get_cnet(batch, self.models, self.extras)
+            cnet_uncond = cnet
             height, width = int(cnet[0].size(-2)*32*4/3), int(cnet[0].size(-1)*32*4/3)
             stage_c_latent_shape, stage_b_latent_shape = calculate_latent_sizes(height, width, batch_size=batch_size)
             # PREPARE CONDITIONS