DeepSeek-Math/cog.yaml

# Configuration for Cog ⚙️
# Reference: https://github.com/replicate/cog/blob/main/docs/yaml.md

build:
  gpu: true
  python_version: "3.11"
  python_packages:
    - torch==2.0.1
    - torchvision==0.15.2
    - transformers==4.37.2
    - accelerate==0.27.0
    - hf_transfer

# predict.py defines how predictions are run on your model
predict: "replicate/predict.py:Predictor"
replicate 2024-02-11 09:45:44 -05:00			`# Configuration for Cog ⚙️`
			`# Reference: https://github.com/replicate/cog/blob/main/docs/yaml.md`

			`build:`
			`gpu: true`
			`python_version: "3.11"`
			`python_packages:`
			`- torch==2.0.1`
			`- torchvision==0.15.2`
			`- transformers==4.37.2`
			`- accelerate==0.27.0`
			`- hf_transfer`

			`# predict.py defines how predictions are run on your model`
replicate 2024-02-12 16:30:24 -05:00			`predict: "replicate/predict.py:Predictor"`