CLIPVisionEncode can now encode multiple images.

2023-08-14 16:54:05 -04:00
parent 0cb6dac943
commit 9cc12c833d
3 changed files with 12 additions and 12 deletions
--- a/comfy/clip_vision.py
+++ b/comfy/clip_vision.py
@@ -24,8 +24,8 @@ class ClipVisionModel():
        return self.model.load_state_dict(sd, strict=False)

    def encode_image(self, image):
-        img = torch.clip((255. * image[0]), 0, 255).round().int()
-        inputs = self.processor(images=[img], return_tensors="pt")
+        img = torch.clip((255. * image), 0, 255).round().int()
+        inputs = self.processor(images=img, return_tensors="pt")
        outputs = self.model(**inputs)
        return outputs