mirror of
https://github.com/storytold/storyteller-ml.git
synced 2026-10-17 00:23:14 +00:00
Brandon Thomas
2d2b25d5a6
MIT license Commit: https://github.com/OpenTalker/SadTalker commit ae0b26359d44c9379450a8e02a38b45eeec15549 (HEAD -> main, origin/main, origin/HEAD) Author: shadowcun <shadowcun@tencent.com> Date: Mon Jul 3 13:48:35 2023 +0800 fixed bug: hacking facexlib of FAN
36 lines
1.1 KiB
YAML
36 lines
1.1 KiB
YAML
build:
|
|
gpu: true
|
|
cuda: "11.3"
|
|
python_version: "3.8"
|
|
system_packages:
|
|
- "ffmpeg"
|
|
- "libgl1-mesa-glx"
|
|
- "libglib2.0-0"
|
|
python_packages:
|
|
- "torch==1.12.1"
|
|
- "torchvision==0.13.1"
|
|
- "torchaudio==0.12.1"
|
|
- "joblib==1.1.0"
|
|
- "scikit-image==0.19.3"
|
|
- "basicsr==1.4.2"
|
|
- "facexlib==0.3.0"
|
|
- "resampy==0.3.1"
|
|
- "pydub==0.25.1"
|
|
- "scipy==1.10.1"
|
|
- "kornia==0.6.8"
|
|
- "face_alignment==1.3.5"
|
|
- "imageio==2.19.3"
|
|
- "imageio-ffmpeg==0.4.7"
|
|
- "librosa==0.9.2" #
|
|
- "tqdm==4.65.0"
|
|
- "yacs==0.1.8"
|
|
- "gfpgan==1.3.8"
|
|
- "dlib-bin==19.24.1"
|
|
- "av==10.0.0"
|
|
- "trimesh==3.9.20"
|
|
run:
|
|
- mkdir -p /root/.cache/torch/hub/checkpoints/ && wget --output-document "/root/.cache/torch/hub/checkpoints/s3fd-619a316812.pth" "https://www.adrianbulat.com/downloads/python-fan/s3fd-619a316812.pth"
|
|
- mkdir -p /root/.cache/torch/hub/checkpoints/ && wget --output-document "/root/.cache/torch/hub/checkpoints/2DFAN4-cd938726ad.zip" "https://www.adrianbulat.com/downloads/python-fan/2DFAN4-cd938726ad.zip"
|
|
|
|
predict: "predict.py:Predictor"
|