cog.yaml

# Configuration for Cog ⚙️
# Reference: https://github.com/replicate/cog/blob/main/docs/yaml.md

build:
  # set to true if your model requires a GPU
  gpu: true
  cuda: "11.8"

  # a list of ubuntu apt packages to install
  #system_packages:
  #  - "libcairo2-dev"
  #  # - "libglib2.0-0"

  # python version in the form '3.8' or '3.8.12'
  python_version: "3.8"

  # a list of packages in the format <package-name>==<version>
  python_packages:
    - "contexttimer==0.3.3"
    - "decord==0.6.0"
    - "einops==0.6.1"
    - "fairscale==0.4.4"
    - "ftfy==6.1.1"
    - "iopath==0.1.10"
    - "ipython==8.13.0"
    - "omegaconf==2.3.0"
    - "opencv-python-headless==4.5.5.64"
    - "opendatasets==0.1.22"
    - "packaging==23.1"
    - "pandas==2.0.1"
    - "plotly==5.14.1"
    - "pre-commit==3.3.2"
    - "pycocoevalcap==1.2"
    - "pycocotools==2.0.6"
    - "python-magic==0.4.27"
    - "scikit-image==0.20.0"
    - "sentencepiece==0.1.99"
    - "spacy==3.5.3"
    - "streamlit==1.22.0"
    - "timm==0.4.12"
    - "torch==2.0.1"
    - "torchvision==0.15.2"
    - "tornado==6.3.2"
    - "tqdm==4.65.0"
    - "transformers==4.29.2"
    - "webdataset==0.2.48"
    - "wheel==0.38.4"

    # - "torch==1.8.0"
    # - "torchvision==0.9.0"
  
  # commands run after the environment is setup
  # run:
    # - "echo env is ready!"
    # - "echo another command if needed"

# predict.py defines how predictions are run on your model
predict: "predict.py:Predictor"