mirror of
https://github.com/deepseek-ai/DeepSeek-VL
synced 2025-02-22 03:37:32 +00:00
20 lines
483 B
YAML
20 lines
483 B
YAML
|
# Configuration for Cog ⚙️
|
||
|
|
# Reference: https://github.com/replicate/cog/blob/main/docs/yaml.md
|
||
|
|
|
||
|
|
build:
|
||
|
|
gpu: true
|
||
|
|
python_version: "3.9"
|
||
|
|
python_packages:
|
||
|
|
- "accelerate==0.27.2"
|
||
|
|
- "attrdict==2.0.1"
|
||
|
|
- "einops==0.7.0"
|
||
|
|
- "sentencepiece==0.2.0"
|
||
|
|
- "torch==2.0.1"
|
||
|
|
- "torchvision==0.15.2"
|
||
|
|
- "transformers>=4.38.2"
|
||
|
|
- "timm>=0.9.16"
|
||
|
|
- "hf_transfer==0.1.6"
|
||
|
|
|
||
|
|
# predict.py defines how predictions are run on your model
|
||
|
|
predict: "predict.py:Predictor"
|