Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/lietorch.git" did not exist on "a1580fe03590f13c1638f8bad88a0d52fa4b6b81"
Commit 4e89b2c2 authored by comfyanonymous's avatar comfyanonymous
Browse files

Put clip vision outputs on the CPU.

parent a094b45c
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 0 deletions
comfy/clip_vision.py
View file @ 4e89b2c2
...@@ -50,6 +50,11 @@ class ClipVisionModel(): ...@@ -50,6 +50,11 @@ class ClipVisionModel():
with precision_scope(comfy.model_management.get_autocast_device(self.load_device), torch.float32): with precision_scope(comfy.model_management.get_autocast_device(self.load_device), torch.float32):
outputs = self.model(pixel_values=pixel_values) outputs = self.model(pixel_values=pixel_values)
for k in outputs:
t = outputs[k]
if t is not None:
outputs[k] = t.cpu()
return outputs return outputs
def convert_to_transformers(sd, prefix): def convert_to_transformers(sd, prefix):
......
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