mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-03-15 14:09:36 +00:00
Don't leave very large hidden states in the clip vision output.
This commit is contained in:
parent
fb3b728203
commit
ed58730658
@ -56,6 +56,7 @@ class ClipVisionModel():
|
|||||||
if t is not None:
|
if t is not None:
|
||||||
if k == 'hidden_states':
|
if k == 'hidden_states':
|
||||||
outputs["penultimate_hidden_states"] = t[-2].cpu()
|
outputs["penultimate_hidden_states"] = t[-2].cpu()
|
||||||
|
outputs["hidden_states"] = None
|
||||||
else:
|
else:
|
||||||
outputs[k] = t.cpu()
|
outputs[k] = t.cpu()
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user