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