pharmapsychotic/clip-interrogator

running demo error

sevenactors opened this issue · 2 comments

I was running the demo code shown in readme. But i get the error below

from PIL import Image
from clip_interrogator import Config, Interrogator
image_path = '001.png'
image = Image.open(image_path).convert('RGB')
ci = Interrogator(Config(clip_model_name="ViT-H-14/laion2b_s32b_b79k"))
print(ci.interrogate(image))

RuntimeError: The size of tensor a (8) must match the size of tensor b (64) at non-singleton dimension 0

I also get this with the ViT-L-14/openai. Not sure what to think. Is it too large?

I also get this with the ViT-L-14/openai. Not sure what to think. Is it too large?

#62