2024-07-06 11:13:39 +00:00
|
|
|
# Configuration for Cog ⚙️
|
|
|
|
|
# Reference: https://cog.run/yaml
|
|
|
|
|
|
|
|
|
|
build:
|
|
|
|
|
gpu: true
|
2024-07-06 17:48:43 +00:00
|
|
|
python_version: "3.10"
|
|
|
|
|
system_packages:
|
|
|
|
|
- "ffmpeg"
|
|
|
|
|
- "libsm6"
|
|
|
|
|
- "libxext6"
|
|
|
|
|
- "libgl1"
|
2024-07-06 11:13:39 +00:00
|
|
|
python_packages:
|
|
|
|
|
- "torch==2.3.0"
|
|
|
|
|
- "torchvision==0.18.0"
|
|
|
|
|
- "torchaudio==2.3.0"
|
|
|
|
|
- "numpy==1.26.4"
|
|
|
|
|
- "pyyaml==6.0.1"
|
|
|
|
|
- "opencv-python==4.10.0.84"
|
|
|
|
|
- "scipy==1.13.1"
|
|
|
|
|
- "imageio==2.34.2"
|
|
|
|
|
- "lmdb==1.4.1"
|
|
|
|
|
- "tqdm==4.66.4"
|
|
|
|
|
- "rich==13.7.1"
|
|
|
|
|
- "ffmpeg==1.4"
|
|
|
|
|
- "onnxruntime-gpu==1.18.0"
|
|
|
|
|
- "onnx==1.16.1"
|
|
|
|
|
- "scikit-image==0.24.0"
|
|
|
|
|
- "albumentations==1.4.10"
|
|
|
|
|
- "matplotlib==3.9.0"
|
|
|
|
|
- "imageio-ffmpeg==0.5.1"
|
|
|
|
|
- "tyro==0.8.5"
|
2024-07-07 13:46:28 +00:00
|
|
|
- "gradio==3.48.0"
|
2024-07-06 11:13:39 +00:00
|
|
|
predict: "predict.py:Predictor"
|
