Update img2panoimg/image_to_360panorama_image_pipeline.py

img2panoimg/image_to_360panorama_image_pipeline.py

@@ -77,7 +77,6 @@ class Image2360PanoramaImagePipeline(DiffusionPipeline):
                 self.pipe.enable_xformers_memory_efficient_attention()
         except Exception as e:
             print(e)
-        self.pipe.enable_model_cpu_offload()
 
         # init controlnet-sr model
         base_model_path = model + '/sr-base'
@@ -96,7 +95,6 @@ class Image2360PanoramaImagePipeline(DiffusionPipeline):
                 self.pipe_sr.enable_xformers_memory_efficient_attention()
         except Exception as e:
             print(e)
-        self.pipe_sr.enable_model_cpu_offload()
         device = torch.device("cuda")
         model_path = model + '/RealESRGAN_x2plus.pth'
         self.upsampler = LazyRealESRGAN(device=device, scale=2)