Don't leave very large hidden states in the clip vision output.

1 year ago · ed58730658
parent fb3b728203
commit ed58730658
1 changed files with 1 additions and 0 deletions
--- a/comfy/clip_vision.py
+++ b/comfy/clip_vision.py
@ -56,6 +56,7 @@ class ClipVisionModel():
            if t is not None:
                if k == 'hidden_states':
                    outputs["penultimate_hidden_states"] = t[-2].cpu()
+                    outputs["hidden_states"] = None
                else:
                    outputs[k] = t.cpu()