File size: 537 Bytes
00e6746 |
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
system_packages:
- "libgl1-mesa-glx"
- "libglib2.0-0"
- "cmake"
python_version: "3.11"
python_packages:
- torch==2.0.1
- torchvision==0.15.2
- lmdb==1.4.1
- opencv_python==4.9.0.80
- Pillow==10.1.0
- PyYAML==6.0.1
- scipy==1.11.4
- timm==0.9.12
- tqdm==4.65.0
- scikit-image==0.22.0
- ipython
run:
- pip install dlib
predict: "predict.py:Predictor"
|