Skip to content

Commit 7ac999b

Browse files
Add image sizes to clip vision outputs. (Comfy-Org#11923)
1 parent 0c6b36c commit 7ac999b

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

comfy/clip_vision.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -66,6 +66,7 @@ def encode_image(self, image, crop=True):
6666
outputs = Output()
6767
outputs["last_hidden_state"] = out[0].to(comfy.model_management.intermediate_device())
6868
outputs["image_embeds"] = out[2].to(comfy.model_management.intermediate_device())
69+
outputs["image_sizes"] = [pixel_values.shape[1:]] * pixel_values.shape[0]
6970
if self.return_all_hidden_states:
7071
all_hs = out[1].to(comfy.model_management.intermediate_device())
7172
outputs["penultimate_hidden_states"] = all_hs[:, -2]

0 commit comments

Comments
 (0)