rom1504 / clip-retrieval

Easily compute clip embeddings and build a clip retrieval system with them
https://rom1504.github.io/clip-retrieval/
MIT License
2.25k stars 203 forks source link

Unable to use clip.load #382

Closed Agaeti closed 2 weeks ago

Agaeti commented 2 weeks ago

Hello,

I'm not sure if I am missing something that should be obvious but I am having trouble replicating the way the clip_retrieval/clip_filter.py file loads the clip model. After using the exact same code starting from the import lines, I get an error at my equivalent of line 20:

AttributeError: module 'clip' has no attribute 'load'

I believe I have all of the same dependencies installed from the requirements.txt. The only difference should be that I moved this code out of the function definition to be global instead and am running this in a Jupyter notebook. If anyone can point me in the right direction for this I would be grateful! Thanks