# Configuration for Cog ⚙️ | |
# Reference: https://github.com/replicate/cog/blob/main/docs/yaml.md | |
build: | |
gpu: true | |
cuda: "11.8" | |
python_version: "3.11" | |
python_packages: | |
- "accelerate==0.25.0" | |
- "diffusers==0.24.0" | |
- "safetensors==0.4.1" | |
- "torch==2.0.1" | |
- "torchvision==0.15.2" | |
- "transformers==4.36.0" | |
run: | |
- curl -o /usr/local/bin/pget -L "https://github.com/replicate/pget/releases/download/v0.3.1/pget" && chmod +x /usr/local/bin/pget | |
# predict.py defines how predictions are run on your model | |
predict: "predict.py:Predictor" | |