File size: 286 Bytes
188fd89 |
1 2 3 4 5 6 7 8 9 10 11 12 |
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"
|