last test vae enabling

handler.py

@@ -24,19 +24,11 @@ class EndpointHandler():
 
 
 
-         def patch_conv(cls):
-                init = cls.__init__
-                def __init__(self, *args, **kwargs):
-                    return init(self, *args, **kwargs, padding_mode='circular')
-                cls.__init__ = __init__
-
-         patch_conv(torch.nn.Conv2d)
-
-
 
 
          self.pipe = StableDiffusionPipeline.from_pretrained(self.stable_diffusion_id,torch_dtype=dtype,safety_checker=StableDiffusionSafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker", torch_dtype=dtype)).to(device.type)
-
+         self.pipe.enable_xformers_memory_efficient_attention()
+         self.pipe.enable_vae_tiling()
          self.generator = torch.Generator(device=device.type).manual_seed(3)
 
      def __call__(self, data: Any) -> List[List[Dict[str, float]]]: