1580158
/
Stable-Diffusion-Webui-Civitai-Helper
mirror of https://github.com/butaixianran/Stable-Diffusion-Webui-Civitai-Helper.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix all image took as nsfw image problem 

nsfw values: "None"、"X"、"Soft"...
pull/174/head
codepiano 2023-05-24 02:05:26 +08:00
parent 920ca3267f
commit 557e06e4b3
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/ch_lib/civitai.py
View File

@ -342,8 +342,8 @@ def get_preview_image_by_model_path(model_path:str, max_size_preview, skip_nsfw_
if model_info["images"]: if model_info["images"]:
for img_dict in model_info["images"]: for img_dict in model_info["images"]:
if "nsfw" in img_dict.keys(): if "nsfw" in img_dict.keys():
if img_dict["nsfw"]: if img_dict["nsfw"] != "" and img_dict["nsfw"] != "None":
util.printD("This image is NSFW") util.printD("This image is NSFW: " + str(img_dict["nsfw"]))
if skip_nsfw_preview: if skip_nsfw_preview:
util.printD("Skip NSFW image") util.printD("Skip NSFW image")
continue continue