Update txt2panoimg/text_to_360panorama_image_pipeline.py

txt2panoimg/text_to_360panorama_image_pipeline.py

@@ -74,7 +74,6 @@ class Text2360PanoramaImagePipeline(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'
@@ -93,7 +92,6 @@ class Text2360PanoramaImagePipeline(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)