youngkingdom/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-19 03:35:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add image sizes to clip vision outputs. (#11923)

Browse Source
This commit is contained in:
comfyanonymous
2026-01-16 20:02:28 -08:00
committed by GitHub
parent 0c6b36c6ac
commit 7ac999bf30
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
comfy/clip_vision.py
View File

@ -66,6 +66,7 @@ class ClipVisionModel():
outputs = Output()
outputs["last_hidden_state"] = out[0].to(comfy.model_management.intermediate_device())
outputs["image_embeds"] = out[2].to(comfy.model_management.intermediate_device())
outputs["image_sizes"] = [pixel_values.shape[1:]] * pixel_values.shape[0]
if self.return_all_hidden_states:
all_hs = out[1].to(comfy.model_management.intermediate_device())
outputs["penultimate_hidden_states"] = all_hs[:, -2]