Closed afiaka87 closed 3 years ago
I've fixed the bug (and added the ability to choose between the ViT-B/32 or RN50 models of CLIP) in this pull request.
Having a little trouble with the CLIP model code. Here's a PR with just the CLI fix (it's one line in cli.py).
Hey, I'm sorry. I introduced this because I changed the text to be optional in my PR. I did not test the CLI and thought someone would check it. It surprised me that the PR was merged directly - but now I know. Next time I'll check more thoroughly!
Hey, I'm sorry. I introduced this because I changed the text to be optional in my PR. I did not test the CLI and thought someone would check it. It surprised me that the PR was merged directly - but now I know. Next time I'll check more thoroughly!
No problem. The versioning probably mitigated it to a degree. Having said that, @lucidrains it might be good to remove the pip --upgrade call in your colab notebook.
Usage:
Output: