From 6f7869f365ca4ec34b017f9002b9dcb9bce56b1a Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Wed, 17 Jul 2024 13:05:38 -0400 Subject: [PATCH] Get clip vision image size from config. --- comfy/clip_vision.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/comfy/clip_vision.py b/comfy/clip_vision.py index acc86be8556..ac39c62279d 100644 --- a/comfy/clip_vision.py +++ b/comfy/clip_vision.py @@ -34,6 +34,7 @@ def __init__(self, json_config): with open(json_config) as f: config = json.load(f) + self.image_size = config.get("image_size", 224) self.load_device = comfy.model_management.text_encoder_device() offload_device = comfy.model_management.text_encoder_offload_device() self.dtype = comfy.model_management.text_encoder_dtype(self.load_device) @@ -50,7 +51,7 @@ def get_sd(self): def encode_image(self, image): comfy.model_management.load_model_gpu(self.patcher) - pixel_values = clip_preprocess(image.to(self.load_device)).float() + pixel_values = clip_preprocess(image.to(self.load_device), size=self.image_size).float() out = self.model(pixel_values=pixel_values, intermediate_output=-2) outputs = Output()