Transformers How to use CompVis/stable-diffusion-safety-checker with Transformers:
# Load model directly
from transformers import AutoProcessor, SafetyChecker
processor = AutoProcessor.from_pretrained("CompVis/stable-diffusion-safety-checker")
model = SafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker", device_map="auto") 
