diffusers/tools

1#!/usr/bin/env python32import PIL3from transformers import CLIPImageProcessor4from diffusers.pipelines.stable_diffusion import StableDiffusionSafetyChecker5 6feature_extractor = CLIPImageProcessor.from_pretrained("runwayml/stable-diffusion-v1-5", subfolder="feature_extractor")7safety_checker = StableDiffusionSafetyChecker.from_pretrained("runwayml/stable-diffusion-v1-5", subfolder="safety_checker")8device = "cpu"9 10image = PIL.Image.open("/home/patrick/images/0.png")11 12safety_checker_input = feature_extractor(image, return_tensors="pt").to(device)13image, has_nsfw_concept = safety_checker(images=image, clip_input=safety_checker_input.pixel_values)14