From ae270f79bcb0da0efe09094eddc488a17a900913 Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Wed, 16 Aug 2023 11:05:11 -0400
Subject: [PATCH] Fix potential issue with batch size and clip vision.

---
 comfy/clip_vision.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/comfy/clip_vision.py b/comfy/clip_vision.py
index 8d04faf..c408a73 100644
--- a/comfy/clip_vision.py
+++ b/comfy/clip_vision.py
@@ -25,6 +25,8 @@ class ClipVisionModel():
 
     def encode_image(self, image):
         img = torch.clip((255. * image), 0, 255).round().int()
+        if len(img.shape) == 3:
+            img = [img]
         inputs = self.processor(images=img, return_tensors="pt")
         outputs = self.model(**inputs)
         return outputs