From ed58730658d0213600b64849d721a6bb92c675bf Mon Sep 17 00:00:00 2001 From: comfyanonymous 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 9b95ae0..1206c68 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()