File size: 628 Bytes
c945057 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
# Configuration for Cog ⚙️
# Reference: https://github.com/replicate/cog/blob/main/docs/yaml.md
build:
gpu: true
cuda: "11.4"
python_version: "3.10"
system_packages:
- "ffmpeg"
- "libsm6"
- "libxext6"
python_packages:
- "diffusers==0.11.1"
- "torch==1.13.0"
- "ftfy==6.1.1"
- "scipy==1.9.3"
- "transformers==4.25.1"
- "accelerate==0.15.0"
- "git+https://github.com/cloneofsimo/lora.git@799c17aef2a475641fb70d68a6992de4fc325ce4"
- "opencv-python==4.7.0.68"
- "torchvision==0.14.0"
- "mediapipe==0.9.0.1"
- "flake8==5.0.4"
predict: "$PREDICT_FILE:Predictor"
|