AIGaming
/

beautiful_realistic_asians

StableDiffusionPipeline

stable-diffusion

Inference Endpoints

Model card Files Files and versions Community

Allen commited on Jun 18

Commit

32d1201

•

1 Parent(s): f132c07

a

Files changed (1) hide show

handler.py +1 -0

handler.py CHANGED Viewed

@@ -17,6 +17,7 @@ class EndpointHandler():
         print("Loading model from:", path)
         self.pipe = StableDiffusionPipeline.from_pretrained(path, torch_dtype=torch.float16)
         self.pipe = self.pipe.to(device)
     def __call__(self, data: Any) -> List[List[Dict[str, float]]]:

         print("Loading model from:", path)
         self.pipe = StableDiffusionPipeline.from_pretrained(path, torch_dtype=torch.float16)
         self.pipe = self.pipe.to(device)
+        self.pipe.safety_checker = lambda images, clip_input: (images, None)
     def __call__(self, data: Any) -> List[List[Dict[str, float]]]: