File size: 414 Bytes
d31a6c6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
# Configuration for Cog ⚙️
# Reference: https://github.com/replicate/cog/blob/main/docs/yaml.md
build:
gpu: true
python_version: "3.10"
cuda: "11.6"
python_packages:
- "torch==1.13.1"
- "sentencepiece==0.1.97"
- "accelerate==0.16.0"
run:
- "pip install git+https://github.com/huggingface/transformers.git@c3dc391da81e6ed7efce42be06413725943b3920"
predict: "predict.py:Predictor"
|