Skip to content

GitLab

"...resnet50_tensorflow.git" did not exist on "418829a20cd16774dce5f0848a4c131b5ab56122"
Commit ed587306 authored by comfyanonymous's avatar comfyanonymous
Browse files

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

parent fb3b7282
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
comfy/clip_vision.py
View file @ ed587306
...@@ -56,6 +56,7 @@ class ClipVisionModel(): ...@@ -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()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment