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"
  run:
  - git clone https://github.com/salesforce/BLIP /root/blip

predict: "predict.py:Predictor"