mirror of
https://github.com/deepseek-ai/Janus.git
synced 2025-02-23 14:18:58 -05:00
25 lines
627 B
YAML
25 lines
627 B
YAML
# Configuration for Cog ⚙️
|
|
# Reference: https://cog.run/yaml
|
|
|
|
build:
|
|
gpu: true
|
|
cuda: "12.1"
|
|
python_version: "3.11"
|
|
python_packages:
|
|
- "torch==2.2"
|
|
- "torchvision"
|
|
- "transformers==4.36.2"
|
|
- "accelerate==1.3.0"
|
|
- "diffusers==0.32.2"
|
|
- "opencv-python==4.10.0.84"
|
|
- "attrdict==2.0.1"
|
|
- "timm==1.0.14"
|
|
- "sentencepiece==0.2.0"
|
|
- "einops==0.8.0"
|
|
- "pillow==10.2.0"
|
|
- "numpy==1.24.3"
|
|
|
|
run:
|
|
- curl -o /usr/local/bin/pget -L "https://github.com/replicate/pget/releases/download/v0.8.2/pget_linux_x86_64" && chmod +x /usr/local/bin/pget
|
|
|
|
predict: "predict.py:Predictor" |