mirror of
https://github.com/THUDM/CogVideo.git
synced 2025-04-06 03:57:56 +08:00
38 lines
1.0 KiB
YAML
38 lines
1.0 KiB
YAML
# Configuration for Cog ⚙️
|
|
# Reference: https://cog.run/yaml
|
|
|
|
build:
|
|
# set to true if your model requires a GPU
|
|
gpu: true
|
|
|
|
# a list of ubuntu apt packages to install
|
|
system_packages:
|
|
- "libgl1-mesa-glx"
|
|
- "libglib2.0-0"
|
|
|
|
# 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:
|
|
- diffusers>=0.30.3
|
|
- accelerate>=0.34.2
|
|
- transformers>=4.44.2
|
|
- numpy==1.26.0
|
|
- torch>=2.4.0
|
|
- torchvision>=0.19.0
|
|
- sentencepiece>=0.2.0
|
|
- SwissArmyTransformer>=0.4.12
|
|
- imageio>=2.35.1
|
|
- imageio-ffmpeg>=0.5.1
|
|
- openai>=1.45.0
|
|
- moviepy>=2.0.0
|
|
- pillow==9.5.0
|
|
- pydantic==1.10.7
|
|
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.py defines how predictions are run on your model
|
|
predict: "predict_t2v.py:Predictor"
|
|
# predict: "predict_i2v.py:Predictor"
|