mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-09-10 19:46:38 +00:00
Fix clip vision issue with old transformers versions.
This commit is contained in:
@@ -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
|
||||
|
Reference in New Issue
Block a user