build:
  gpu: true
  cuda: "11.6.2"
  python_version: "3.10"
  python_packages:
    - "diffusers==0.2.4"
    - "torch==1.12.1 --extra-index-url=https://download.pytorch.org/whl/cu116"
    - "ftfy==6.1.1"
    - "scipy==1.9.0"
    - "transformers==4.21.1"
predict: "predict.py:Predictor"