Daniil4rewr32r23 commited on
Commit
053c713
1 Parent(s): b5d3ac5

Update inpainting.py

Browse files
Files changed (1) hide show
  1. inpainting.py +0 -4
inpainting.py CHANGED
@@ -184,10 +184,6 @@ class StableDiffusionInpaintingPipeline(DiffusionPipeline):
184
  image = (image / 2 + 0.5).clamp(0, 1)
185
  image = image.cpu().permute(0, 2, 3, 1).numpy()
186
 
187
- # run safety checker
188
- safety_cheker_input = self.feature_extractor(self.numpy_to_pil(image), return_tensors="pt").to(self.device)
189
- image, has_nsfw_concept = self.safety_checker(images=image, clip_input=safety_cheker_input.pixel_values)
190
-
191
  if output_type == "pil":
192
  image = self.numpy_to_pil(image)
193
 
 
184
  image = (image / 2 + 0.5).clamp(0, 1)
185
  image = image.cpu().permute(0, 2, 3, 1).numpy()
186
 
 
 
 
 
187
  if output_type == "pil":
188
  image = self.numpy_to_pil(image)
189