nsfw模型下载
https://huggingface.co/AdamCodd/vit-base-nsfw-detector/resolve/main/config.json
https://huggingface.co/AdamCodd/vit-base-nsfw-detector/resolve/main/model.safetensors
https://huggingface.co/AdamCodd/vit-base-nsfw-detector/resolve/main/preprocessor_config.json
下载放到对应的/models/nsfw_detector/vit-base-nsfw-detector目录下面
网盘下载地址:
链接:https://pan.quark.cn/s/3960d3e721a5
提取码:wkF2
如何关闭nsfw
修改reactor_sfw.py:文件地址:/custom_nodes/comfyui-reactor/scripts/reactor_sfw.py
修改代码如下:
from transformers import pipeline
from PIL import Image
import io
import logging
import os
import comfy.model_management as model_management
from reactor_utils import download
from scripts.reactor_logger import logger
MODEL_EXISTS = False
def ensure_nsfw_model(nsfwdet_model_path):
"""Download NSFW detection model if it doesn't exist"""
global MODEL_EXISTS
MODEL_EXISTS = True
return True
SCORE = 0.979
logging.getLogger("transformers").setLevel(logging.ERROR)
def nsfw_image(img_data, model_path: str):
return False无nsfw限制版本下载地址:
v0.6.1
链接:https://pan.quark.cn/s/8cbca9fba163
提取码:qg69
v0.7.1
链接:https://pan.quark.cn/s/47809e632d18
提取码:AvyN