Spaces:
Runtime error
Runtime error
remove the safely checker
#4
by
hellcat82
- opened
- inpainting.py +2 -3
inpainting.py
CHANGED
|
@@ -184,9 +184,8 @@ class StableDiffusionInpaintingPipeline(DiffusionPipeline):
|
|
| 184 |
image = (image / 2 + 0.5).clamp(0, 1)
|
| 185 |
image = image.cpu().permute(0, 2, 3, 1).numpy()
|
| 186 |
|
| 187 |
-
|
| 188 |
-
|
| 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)
|
|
|
|
| 184 |
image = (image / 2 + 0.5).clamp(0, 1)
|
| 185 |
image = image.cpu().permute(0, 2, 3, 1).numpy()
|
| 186 |
|
| 187 |
+
|
| 188 |
+
|
|
|
|
| 189 |
|
| 190 |
if output_type == "pil":
|
| 191 |
image = self.numpy_to_pil(image)
|