Remove compile for stage2 and 3 (#139)

- Remove compile for stage2 and 3 (bcc86c65b32f3751244c857b98be741b15ac7d7f)


Co-authored-by: Patrick von Platen <[email protected]>

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()