| 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" |
|
|
