Remove compile for stage2 and 3

model.py

@@ -63,13 +63,8 @@ class Model:
             self.pipe.unet.to(memory_format=torch.channels_last)
             self.pipe.unet = torch.compile(self.pipe.unet, mode="reduce-overhead", fullgraph=True)
 
-            self.super_res_1_pipe.unet.to(memory_format=torch.channels_last)
-            self.super_res_1_pipe.unet = torch.compile(self.super_res_1_pipe.unet, mode="reduce-overhead", fullgraph=True)
-
             if not DISABLE_SD_X4_UPSCALER:
                 self.super_res_2_pipe.to(self.device)
-                self.super_res_2_pipe.unet.to(memory_format=torch.channels_last)
-                self.super_res_2_pipe.unet = torch.compile(self.super_res_2_pipe.unet, mode="reduce-overhead", fullgraph=True)
         else:
             self.pipe.enable_model_cpu_offload()
             self.super_res_1_pipe.enable_model_cpu_offload()