| build: | |
| gpu: true | |
| cuda: "11.8" | |
| python_version: "3.11.1" | |
| python_packages: | |
| - "diffusers==0.11.1" | |
| - "torch==1.13.0" | |
| - "ftfy==6.1.1" | |
| - "scipy==1.9.3" | |
| - "transformers==4.25.1" | |
| - "accelerate==0.15.0" | |
| - "huggingface-hub==0.13.2" | |
| predict: "predict.py:Predictor" | |