Skip to content

GitLab

"...composable_kernel_onnx.git" did not exist on "9d99a5807298c3f263d39a08328c3c68c930a900"
Commit 58f0c616 authored by comfyanonymous's avatar comfyanonymous
Browse files

Fix clip vision issue with old transformers versions.

parent ae270f79
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
comfy/clip_vision.py
View file @ 58f0c616
...@@ -25,8 +25,7 @@ class ClipVisionModel(): ...@@ -25,8 +25,7 @@ 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 = list(map(lambda a: a, img))
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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment