CompVis / stable-diffusion

A latent text-to-image diffusion model
https://ommer-lab.com/research/latent-diffusion-models/
Other
66.54k stars 9.97k forks source link

Fine-Tuning script for NSFW filter #783

Open CreativeBuilds opened 11 months ago

CreativeBuilds commented 11 months ago

I've combed through the issues in the repo and have seen tons of requests asking to turn off the nsfw filter, however this option is no viable for me as I'm looking to run a service on top of Stable Diffusion.

However the amount of false positives that come about are simply too great. I was wondering if there was a script somewhere from CompVis directly, that allows you to fine tune the CompVis/stable-diffusion-safety-checker model.

Or a description on how I'd go about doing that would be great. Thanks!