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