ComfyUI Extension: ImageReward

Authored by ZaneA

Created

Updated

28 stars

NODES: ImageRewardLoader, ImageRewardScore

README

ComfyUI-ImageReward

Score images with ImageReward.

Installation

Within your ComfyUI directory, and within your Python venv:

cd custom_nodes
git clone https://github.com/ZaneA/ComfyUI-ImageReward
pip install -r ComfyUI-ImageReward/requirements.txt

Usage

Add an ImageRewardLoader node, this has the default model name prefilled, and is passed directly to the ImageReward loader. Note, the model will be downloaded on first run.

Add an ImageRewardScore node, connect the model, your image, and your prompt (either enter this directly, or right click the node and convert prompt to an input first). Connect the SCORE_FLOAT or SCORE_STRING output to an appropriate node.

Node screenshot