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