Compare commits

..

1 Commits

Author SHA1 Message Date
B. Bergeron
cbc17e3156
Merge fa71050a07 into 0c6b36c6ac 2026-01-16 15:30:39 -07: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]