You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
20 lines
575 B
20 lines
575 B
build: |
|
gpu: true |
|
cuda: "11.6" |
|
python_version: "3.8" |
|
system_packages: |
|
- "libgl1-mesa-glx" |
|
- "libglib2.0-0" |
|
python_packages: |
|
- "ipython==8.4.0" |
|
- "fairscale==0.4.12" |
|
- "transformers==4.21.2" |
|
- "ftfy==6.1.1" |
|
- "torch==1.13.0 --extra-index-url=https://download.pytorch.org/whl/cu116" |
|
- "torchvision==0.14.0 --extra-index-url=https://download.pytorch.org/whl/cu116" |
|
- "open_clip_torch==2.7.0" |
|
- "timm==0.4.12" |
|
- "pycocoevalcap==1.2" |
|
- "git+https://github.com/pharmapsychotic/BLIP.git" |
|
|
|
predict: "predict.py:Predictor"
|
|
|