mirror of
https://github.com/salesforce/BLIP.git
synced 2026-01-26 15:19:44 +00:00
18 lines
350 B
YAML
18 lines
350 B
YAML
build:
|
|
gpu: true
|
|
cuda: "11.1"
|
|
python_version: "3.8"
|
|
system_packages:
|
|
- "libgl1-mesa-glx"
|
|
- "libglib2.0-0"
|
|
python_packages:
|
|
- "ipython==7.30.1"
|
|
- "torchvision==0.11.1"
|
|
- "torch==1.10.0"
|
|
- "timm==0.4.12"
|
|
- "transformers==4.15.0"
|
|
- "fairscale==0.4.4"
|
|
- "pycocoevalcap==1.2"
|
|
|
|
predict: "predict.py:Predictor"
|