File size: 290 Bytes
7ee4d50 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
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"
|