|
|
|
|
|
|
|
build:
|
|
gpu: true
|
|
cuda: "12.1"
|
|
system_packages:
|
|
- "ffmpeg"
|
|
- "libgl1"
|
|
python_version: "3.10.13"
|
|
python_packages:
|
|
- "torch==2.2.2"
|
|
- "torchvision"
|
|
- "triton==2.2.0"
|
|
- "diffusers==0.11.1"
|
|
- "transformers==4.38.0"
|
|
- "huggingface-hub==0.25.2"
|
|
- "imageio==2.27.0"
|
|
- "decord==0.6.0"
|
|
- "accelerate==0.26.1"
|
|
- "einops==0.7.0"
|
|
- "omegaconf==2.3.0"
|
|
- "safetensors==0.4.2"
|
|
- "opencv-python==4.9.0.80"
|
|
- "mediapipe==0.10.11"
|
|
- "av==11.0.0"
|
|
- "torch-fidelity==0.3.0"
|
|
- "torchmetrics==1.3.1"
|
|
- "python_speech_features==0.6"
|
|
- "librosa==0.10.1"
|
|
- "scenedetect==0.6.1"
|
|
- "ffmpeg-python==0.2.0"
|
|
- "lpips==0.1.4"
|
|
- "face-alignment==1.4.1"
|
|
- "ninja==1.11.1.1"
|
|
- "pandas==2.0.3"
|
|
- "numpy==1.24.4"
|
|
- "xformers==0.0.26"
|
|
|
|
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"
|
|
|
