File size: 764 Bytes
8d66e99 2558496 8d66e99 2558496 8d66e99 4392dfe 8d66e99 2558496 8d66e99 4392dfe 8d66e99 4392dfe 8d66e99 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
# Configuration for Cog
build:
# Use a GPU because PULSE relies on a vision tower and GPU acceleration
gpu: true
python_version: "3.11"
python_packages:
# Core libraries
- "torch==2.1.2"
- "torchvision==0.16.2"
- "transformers==4.37.2"
- "tokenizers==0.15.1"
- "sentencepiece==0.1.99"
- "safetensors>=0.4.2"
- "accelerate==0.30.0"
# Vision and utility libraries
- "timm==0.6.13"
- "pillow>=9.5.0"
- "requests>=2.31.0"
- "numpy"
- "einops==0.6.1"
- "einops-exts==0.0.4"
- "huggingface_hub==0.13.4"
# LLaVA library pulled from GitHub at a fixed tag
- "llava @ git+https://github.com/haotian-liu/LLaVA@v1.2.0"
# The predictor defines how to run inference
predict: "predict.py:Predictor" |