File size: 576 Bytes
2ba7d38 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
# 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"
|