[Bug fix] Fix img2img processor with safety checker (huggingface#3127)

Fix img2img processor with safety checker

Author: patrickvonplaten

pipelines/stable_diffusion/safety_checker.py

@@ -85,7 +85,10 @@ def forward(self, clip_input, images):
 
         for idx, has_nsfw_concept in enumerate(has_nsfw_concepts):
             if has_nsfw_concept:
-                images[idx] = np.zeros(images[idx].shape)  # black image
+                if torch.is_tensor(images) or torch.is_tensor(images[0]):
+                    images[idx] = torch.zeros_like(images[idx])  # black image
+                else:
+                    images[idx] = np.zeros(images[idx].shape)  # black image
 
         if any(has_nsfw_concepts):
             logger.warning(