| # Configuration for Cog ⚙️ | |
| # Reference: https://cog.run/yaml | |
| build: | |
| # set to true if your model requires a GPU | |
| gpu: true | |
| # a list of ubuntu apt packages to install | |
| system_packages: | |
| - "libsndfile1" | |
| - "ffmpeg" | |
| # python version in the form '3.11' or '3.11.4' | |
| python_version: "3.11" | |
| # path to a Python requirements.txt file | |
| python_requirements: requirements.txt | |
| # cog_runtime: true | |
| run: | |
| - curl -o /usr/local/bin/pget -L "https://github.com/replicate/pget/releases/download/v0.8.2/pget_linux_x86_64" && chmod +x /usr/local/bin/pget | |
| # predict.py defines how predictions are run on your model | |
| predict: "predict.py:Predictor" | |