| # Configuration for Cog ⚙️ | |
| # Reference: https://cog.run/yaml | |
| build: | |
| # Set to true if your model requires a GPU | |
| gpu: false | |
| # Python version in the form '3.11' or '3.11.4' | |
| python_version: "3.11" | |
| # A list of packages in the format <package-name>==<version> | |
| python_packages: | |
| - "transformers==4.48.2" | |
| - "datasets==3.2.0" | |
| - "torch==2.5.1" | |
| - "accelerate==1.3.0" | |
| - "sentencepiece==0.2.0" | |
| # predict.py defines how predictions are run on your model | |
| predict: "predict.py:Predictor" | |