build:
  gpu: true
  cuda: "11.3"
  python_version: 3.8
  system_packages:
    - libgl1-mesa-glx
    - libglib2.0-0
  python_packages:
    - torch==1.12.1 --extra-index-url https://download.pytorch.org/whl/cu113
#    - torchvision==0.13.1+cu113--extra-index-url https://download.pytorch.org/whl/cu113
    - albumentations==1.3.0
    - open-clip-torch==2.7.0
    - opencv-python==4.6.0.66
    - imageio==2.22.2
    - opencv-python==4.6.0.66
    - imageio-ffmpeg==0.4.2
    - pytorch-lightning==1.7.7
    - omegaconf==2.1.1
    - test-tube==0.7.5
    - einops==0.3.0
    - transformers==4.19.2
    - webdataset==0.2.5
    - kornia==0.6
#    - open_clip_torch==2.0.2
    - invisible-watermark==0.1.5
#    - streamlit-drawable-canvas==0.8.0
    - torchmetrics==0.10.2


predict: "scripts/replicate/predict.py:Predictor"