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

2025-01-25 15:55:18 +00:00 · 2023-09-12 15:09:10 -04:00 · 2023-09-12 15:09:10 -04:00 · ed58730658
commit ed58730658
parent fb3b728203
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()