Charles-Elena
/

ControlNet-endpoint-CRL-test

stable-diffusion

stable-diffusion-diffusers

endpoints-template

Model card Files Files and versions

yamildiego commited on Feb 21, 2024

Commit

63859a4

·

1 Parent(s): 17fe0cc

old version

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -27,8 +27,8 @@ class EndpointHandler():
          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]]]:

          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]]]: