Update main.py

main.py

@@ -46,13 +46,13 @@ class Upscale_CaseCade:
         self.extras_b.sampling_configs['shift'] = 1
         self.extras_b.sampling_configs['timesteps'] = 10
         self.extras_b.sampling_configs['t_start'] = 1.0
-        self.models = ControlNetCore.Models(
-           **{**self.models.to_dict(), 'generator': torch.compile(self.models.generator, mode="reduce-overhead", fullgraph=True)}
-        )
+        # self.models = ControlNetCore.Models(
+        #    **{**self.models.to_dict(), 'generator': torch.compile(self.models.generator, mode="reduce-overhead", fullgraph=True)}
+        # )
 
-        self.models_b = WurstCoreB.Models(
-           **{**self.models_b.to_dict(), 'generator': torch.compile(self.models_b.generator, mode="reduce-overhead", fullgraph=True)}
-           )
+        # self.models_b = WurstCoreB.Models(
+        #    **{**self.models_b.to_dict(), 'generator': torch.compile(self.models_b.generator, mode="reduce-overhead", fullgraph=True)}
+        #    )