Compare commits

..

1 Commits

Author SHA1 Message Date
Sipovec
2ae5f53f3d
Merge 3e34344eeb into 0c6b36c6ac 2026-01-17 08:21:09 +09:00

View File

@ -66,7 +66,6 @@ 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]