How to disable safe_checker in run_app and run_music_video_app - Githubissues

nateraw / stable-diffusion-videos

Create 🔥 videos with Stable Diffusion by exploring the latent space and morphing between text prompts

Apache License 2.0

4.34k stars 413 forks source link

How to disable safe_checker in run_app and run_music_video_app #193

Closed SoftologyPro closed 1 year ago

SoftologyPro commented 1 year ago

How can I disable the NSFW checking in the run_app.py and run_music_video_app.py scripts? It detects NSFW in safe frames and leads to flickering black frames in output videos which is unwanted.

SoftologyPro commented 1 year ago

Never mind. Worked it out.

pipeline = StableDiffusionWalkPipeline.from_pretrained(
    "CompVis/stable-diffusion-v1-4",
    torch_dtype=torch.float16,
    safety_checker = None,
    requires_safety_checker = False,
).to("cuda")