extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-02-24 22:31:29 +01:00
Code Issues Packages Projects Releases Wiki Activity

Filter NSFW images (if enabled) during live preview

Browse Source
This commit is contained in:
cmdr2 2023-03-16 11:30:16 +05:30
parent 3507f91090
commit ca99b87319
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
ui/easydiffusion/renderer.py
View File

@ -146,6 +146,9 @@ def make_step_callback(
def update_temp_img(x_samples, task_temp_images: list): def update_temp_img(x_samples, task_temp_images: list):
partial_images = [] partial_images = []
images = latent_samples_to_images(context, x_samples) images = latent_samples_to_images(context, x_samples)
if task_data.block_nsfw:
images = apply_filters(context, "nsfw_checker", images)
for i, img in enumerate(images): for i, img in enumerate(images):
buf = img_to_buffer(img, output_format="JPEG") buf = img_to_buffer(img, output_format="JPEG")