Update handler.py

handler.py

@@ -50,14 +50,14 @@ class EndpointHandler():
             image = None
             mask_image = None 
 
-        self.pipe.enable_xformers_memory_efficient_attention()
+        #self.pipe.enable_xformers_memory_efficient_attention()
         
         # run inference pipeline
         out = self.pipe(prompt=prompt, image=image, mask_image=mask_image)
 
         image = out.images[0].resize((1024, 1024))
 
-        self.pipe2.enable_xformers_memory_efficient_attention()
+        #self.pipe2.enable_xformers_memory_efficient_attention()
 
         image = pipe(
             prompt=prompt,
@@ -71,7 +71,7 @@ class EndpointHandler():
         ).images[0]
         
         pipe = AutoPipelineForImage2Image.from_pipe(pipe)
-        self.pipe3.enable_xformers_memory_efficient_attention()
+        #self.pipe3.enable_xformers_memory_efficient_attention()
         
         image = pipe(
             prompt=prompt,