From 58f0c616ed168b728c93d095f2e11041c6de0582 Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Wed, 16 Aug 2023 11:36:22 -0400
Subject: [PATCH] Fix clip vision issue with old transformers versions.

---
 comfy/clip_vision.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

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