Sign in

Nodes Browser

ComfyDeploy: How ComfyUI Safety Checker works in ComfyUI?

What is ComfyUI Safety Checker?

This project provides custom safety checkers for image classification using Falcons AI and CompVis models. The safety checkers are designed to detect and filter out NSFW content from images.

How to install it in ComfyDeploy?

Head over to the machine page

  1. Click on the "Create a new machine" button
  2. Select the Edit build steps
  3. Add a new step -> Custom Node
  4. Search for ComfyUI Safety Checker and select it
  5. Close the build step dialig and then click on the "Save" button to rebuild the machine

ComfyUI Safety Checker

This project provides custom safety checkers for image classification using Falcons AI and CompVis models. The safety checkers are designed to detect and filter out NSFW content from images.

Features

  • Falcons AI Safety Checker
  • CompVis Safety Checker

Requirements

  • Python 3.8+
  • PyTorch
  • Transformers
  • ComfyUI

Installation

  1. Clone the repository:

    git clone https://github.com/shabri-arrahim/ComfyUI-Safety-Checker.git
cd ComfyUI-Safety-Checker

  2. Install the required dependencies:

    pip install -r requirements.txt

  3. Download the models and place them into the diffusers folder in the models directory:

    • Falcon AI model:

      huggingface-cli download Falconsai/nsfw_image_detection --local-dir models/diffusers/Falconsai_nsfw_image_detection

    • CompVis model:

      huggingface-cli download CompVis/stable-diffusion-safety-checker --local-dir models/diffusers/CompVis_stable_diffusion_safety_checker

Usage

Running with ComfyUI

  1. Ensure you have ComfyUI installed and set up. If not, follow the instructions on the ComfyUI GitHub page.

  2. Place this cloned repository into the ComfyUI custom nodes directory:

    cp -r /path/to/ComfyUI-Safety-Checker /path/to/comfyui/custom_nodes/

  3. Start ComfyUI:

    cd /path/to/comfyui
python main.py

  4. In the ComfyUI interface, you should now see the custom nodes FalconsAISafetyChecker and CompVisSafetyChecker available under the "SafetyChecker" category.

  5. Use these nodes in your ComfyUI workflows to filter images for NSFW content.

License

This project is licensed under the MIT License.