nodes/Gourieff-ComfyUI-ReActor
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

FIX: PR30 'model_path' variable name error

Browse Source
This commit is contained in:
Gourieff | 古仁
2025-02-21 18:30:53 +07:00
parent fc69c2232d
commit 5944d9a3e1
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/reactor_sfw.py
+4 -5
View File
@@ -4,19 +4,18 @@ import logging
import os
from reactor_utils import download
def ensure_nsfw_model(model_path):
def ensure_nsfw_model(nsfwdet_model_path):
"""Download NSFW detection model if it doesn't exist"""
if not os.path.exists(model_path):
os.makedirs(model_path)
if not os.path.exists(nsfwdet_model_path):
os.makedirs(nsfwdet_model_path)
nd_urls = [
"https://huggingface.co/AdamCodd/vit-base-nsfw-detector/resolve/main/config.json",
"https://huggingface.co/AdamCodd/vit-base-nsfw-detector/resolve/main/confusion_matrix.png",
"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",
]
for model_url in nd_urls:
model_name = os.path.basename(model_url)
model_path = os.path.join(model_path, model_name)
model_path = os.path.join(nsfwdet_model_path, model_name)
download(model_url, model_path, model_name)
SCORE = 0.965 # 0.965 and less - is safety content