File size: 240 Bytes
383292f |
1 2 3 4 5 6 7 8 9 10 11 12 |
build:
gpu: true
python_version: "3.10"
system_packages:
- "libgl1-mesa-glx"
- "libglib2.0-0"
python_packages:
- "torch==2.0.1"
- "transformers==4.30.2"
- "peft==0.4.0"
predict: "predict.py:Predictor" |