Merge branch 'hotfix/prevent-skipping-and-stopping-by-other-users'

# Conflicts:
#	modules/advanced_parameters.py
#	modules/async_worker.py
#	webui.py
This commit is contained in:
Manuel Schmid
2024-01-22 23:15:18 +01:00
9 changed files with 222 additions and 208 deletions
-5
View File
@@ -1,8 +1,5 @@
# modified version of https://github.com/AUTOMATIC1111/stable-diffusion-webui-nsfw-censor/blob/master/scripts/censor.py
import numpy as np
import torch
import modules.core as core
from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker
from transformers import AutoFeatureExtractor
@@ -16,8 +13,6 @@ safety_checker = None
def numpy_to_pil(image):
image = (image * 255).round().astype("uint8")
#pil_image = Image.fromarray(image, 'RGB')
pil_image = Image.fromarray(image)
return pil_image