From ed58730658d0213600b64849d721a6bb92c675bf Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Tue, 12 Sep 2023 15:09:10 -0400
Subject: [PATCH] Don't leave very large hidden states in the clip vision
output.
---
comfy/clip_vision.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/comfy/clip_vision.py b/comfy/clip_vision.py
index 9b95ae00..1206c680 100644
--- 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()