omni-zero / cog.yaml
root
cog fixes
00c2c78
# 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
# a list of ubuntu apt packages to install
system_packages:
- "libgl1-mesa-glx"
- "libglib2.0-0"
# python version in the form '3.8' or '3.8.12'
python_version: "3.11"
python_requirements: "requirements.txt"
# a list of packages in the format <package-name>==<version>
# python_packages:
# - "numpy==1.19.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"