LAVIS
LAVIS copied to clipboard
Fix requirements file
The requirements file was not updated, casing a few small issues when running pip install -e .
and trying to run the demo.
-
torchvision
version is not specified, thus pip tries to install the latest version which is not compatible with torch 1.10.0, addedtorchvision==0.11.1
to the requirements file which is the compatible version. - The script
lavis/models/img2prompt_models/img2prompt_vqa.py
requiresspacy
, but it was not reflected in the requirements file resulting in an import error when running the demo, addedspacy
to the requirement file. -
spacy
requires a specific version ofpydantic
, see issue inspacy
repo here, addedpydantic==1.10.2
to the requirements file.
Overall small changes, but hopefully it will save a 10 minutes of debugging to the next person:)
Thanks for the contribution! Before we can merge this, we need @GilLevi to sign the Salesforce.com Contributor License Agreement.
This is weird, I signed the CLA, but the PR still has the cls:missing label.
This is weird, I signed the CLA, but the PR still has the cls:missing label.
Nah, it seems to work like this all the time
Thanks @bottledmind, hope it will get merged then.
The VQA pipeline still requires an en_core_web_sm model. I've created https://github.com/salesforce/LAVIS/pull/77 with a hacky way of solving this issue. If you have a better idea, merge it to this PR and inform me to remove mine