/ComfyUI-PickScore-Nodes

PickScore nodes for ComfyUI.

Primary LanguagePythonMIT LicenseMIT

ComfyUI PickScore Nodes

Image scoring nodes for ComfyUI using PickScore to predict which images in a batch best match a given prompt.

Installation

Clone the repository to custom_nodes:

git clone https://github.com/Zuellni/ComfyUI-PickScore-Nodes custom_nodes/ComfyUI-PickScore-Nodes

Nodes

Name Description
Loader Loads scoring models from Hugging Face or a given directory. The default model is about 4 GB.
Processor Takes images/text and converts them to inputs for the Selector node.
Selector Selects up to limit images and passes them to other nodes along with latents/masks of the same shape. Interrupts processing if the threshold isn't reached.

Workflow

An example workflow is embedded in the image below and can be opened in ComfyUI.

workflow