298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-03-15 14:09:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix potential issue with batch size and clip vision.

Browse Source
This commit is contained in:
comfyanonymous 2023-08-16 11:05:11 -04:00
parent 27b87c25a1
commit ae270f79bc
1 changed files with 2 additions and 0 deletions

2
comfy/clip_vision.py
View File

@ -25,6 +25,8 @@ class ClipVisionModel():
def encode_image(self, image): def encode_image(self, image):
img = torch.clip((255. * image), 0, 255).round().int() img = torch.clip((255. * image), 0, 255).round().int()
if len(img.shape) == 3:
img = [img]
inputs = self.processor(images=img, return_tensors="pt") inputs = self.processor(images=img, return_tensors="pt")
outputs = self.model(**inputs) outputs = self.model(**inputs)
return outputs return outputs