diff --git a/README.md b/README.md index 2a38630..b85117c 100644 --- a/README.md +++ b/README.md @@ -61,6 +61,8 @@ The extension adds the following nodes to ComfyUI: This is the main node implementing the random search and zero-order optimization algorithms from the paper. +![InferenceTimeScaler Node](assets/inference_scaler_node.jpeg) + #### Inputs: - `model`: (MODEL) Denoising model - `seed`: (INT) Random seed for reproducibility @@ -96,6 +98,8 @@ This is the main node implementing the random search and zero-order optimization Loads the Qwen VLM verifier model for image evaluation. +![Qwen VLM Verifier Node](assets/qwen_verifier_node.jpeg) + #### Inputs: - `qwen_verifier_id`: Model identifier (default: "Qwen/Qwen2.5-VL-7B-Instruct") - `device`: Device to load model on ("cuda" or "cpu") @@ -107,6 +111,8 @@ Loads the Qwen VLM verifier model for image evaluation. Loads the CLIP model for computing text-image similarity scores. +![CLIP Score Verifier Node](assets/clip_verifier_node.jpeg) + #### Inputs: - `clip_verifier_id`: Model identifier (default: "openai/clip-vit-base-patch32") - `device`: Device to load model on ("cuda" or "cpu") @@ -118,6 +124,8 @@ Loads the CLIP model for computing text-image similarity scores. Loads the ImageReward model for image quality assessment. +![ImageReward Verifier Node](assets/image_reward_node.jpeg) + #### Inputs: - `ir_verifier_id`: Model identifier (default: "ImageReward-v1.0") - `device`: Device to load model on ("cuda" or "cpu") diff --git a/assets/clip_verifier_node.jpeg b/assets/clip_verifier_node.jpeg new file mode 100644 index 0000000..e230e6c Binary files /dev/null and b/assets/clip_verifier_node.jpeg differ diff --git a/assets/image_reward_node.jpeg b/assets/image_reward_node.jpeg new file mode 100644 index 0000000..9cc7502 Binary files /dev/null and b/assets/image_reward_node.jpeg differ diff --git a/assets/inference_scaler_node.jpeg b/assets/inference_scaler_node.jpeg new file mode 100644 index 0000000..7cf13f0 Binary files /dev/null and b/assets/inference_scaler_node.jpeg differ diff --git a/assets/qwen_verifier_node.jpeg b/assets/qwen_verifier_node.jpeg new file mode 100644 index 0000000..f91b26a Binary files /dev/null and b/assets/qwen_verifier_node.jpeg differ