| # 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" | |