File size: 634 Bytes
2cd560a |
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 27 28 |
# Configuration for Cog ⚙️
# Reference: https://github.com/replicate/cog/blob/main/docs/yaml.md
build:
gpu: true
cuda: "11.7"
system_packages:
- "libgl1-mesa-glx"
- "libglib2.0-0"
python_version: "3.10"
python_packages:
- "timm==0.9.2"
- "transformers==4.30.2"
- "fairscale==0.4.13"
- "pycocoevalcap==1.2"
- "torch==1.13.0"
- "torchvision==0.14.0"
- "Pillow==9.5.0"
- "scipy==1.10.1"
- "opencv-python==4.7.0.72"
- "addict==2.4.0"
- "yapf==0.40.0"
- "supervision==0.10.0"
- git+https://github.com/openai/CLIP.git
- ipython
predict: "predict.py:Predictor"
|