feat: add nsfw image censoring via config and checkbox (#958)

Browse Source

* add nsfw image censoring

activatable via config, uses CompVis/stable-diffusion-safety-checker

* fix progressbar call for nsfw output

* use config to set cache dir for safety checker

* add checkbox black_out_nsfw

makes both enabling via config and checkbox possible, where config overrides the checkbox value

* fix: add missing diffusers package

* feat: extract safety checker, remove dependency to diffusers

* feat: make code compatible again after merge with main

* feat: move censor to extras, optimize safety checker file handling

* refactor: rename folder safety_checker_models to safety_checker

...

This commit is contained in:

Manuel Schmid

2024-05-18 15:50:28 +02:00

committed by GitHub

parent 33fa175bd4

commit 00d3d1b4b3

9 changed files with 424 additions and 11 deletions

Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL

Download Patch File Download Diff File Expand all files Collapse all files

models/safety_checker/put_safety_checker_models_here