extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-06-20 18:08:00 +02:00
Code Issues Packages Projects Releases Wiki Activity

Server-side config to allow force-blocking of NSFW images

Browse Source
This commit is contained in:
cmdr2 2023-08-30 16:13:10 +05:30
parent 56bea46e3a
commit 9a9f6e3559
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
ui/easydiffusion/tasks/render_images.py
View File

@ -42,9 +42,16 @@ class RenderTask(Task):
def run(self): def run(self):
"Runs the image generation task on the assigned thread" "Runs the image generation task on the assigned thread"
from easydiffusion import task_manager from easydiffusion import task_manager, app
context = runtime.context context = runtime.context
config = app.getConfig()
if config.get("block_nsfw", False): # override if set on the server
self.task_data.block_nsfw = True
if "nsfw_checker" not in self.task_data.filters:
self.task_data.filters.append("nsfw_checker")
self.models_data.model_paths["nsfw_checker"] = "nsfw_checker"
def step_callback(): def step_callback():
task_manager.keep_task_alive(self) task_manager.keep_task_alive(self)