Commit 05168e5d authored by Patrick von Platen's avatar Patrick von Platen
Browse files

make style

parent 948022e1
...@@ -349,7 +349,8 @@ class StableDiffusionPipelineSafe(DiffusionPipeline): ...@@ -349,7 +349,8 @@ class StableDiffusionPipelineSafe(DiffusionPipeline):
flagged_images = np.zeros((2, *image.shape[1:])) flagged_images = np.zeros((2, *image.shape[1:]))
if any(has_nsfw_concept): if any(has_nsfw_concept):
logger.warning( logger.warning(
"Potential NSFW content was detected in one or more images. A black image will be returned instead." "Potential NSFW content was detected in one or more images. A black image will be returned"
" instead."
f"{'You may look at this images in the `unsafe_images` variable of the output at your own discretion.' if enable_safety_guidance else 'Try again with a different prompt and/or seed.'}" f"{'You may look at this images in the `unsafe_images` variable of the output at your own discretion.' if enable_safety_guidance else 'Try again with a different prompt and/or seed.'}"
) )
for idx, has_nsfw_concept in enumerate(has_nsfw_concept): for idx, has_nsfw_concept in enumerate(has_nsfw_concept):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment