#!/bin/bash
# Create the checkpoints directory
mkdir -p ./checkpoints
# Download model files into the checkpoints directory
wget -nc "https://github.com/OpenTalker/SadTalker/releases/download/v0.0.2-rc/mapping_00109-model.pth.tar" -O "./checkpoints/mapping_00109-model.pth.tar"
wget -nc "https://github.com/OpenTalker/SadTalker/releases/download/v0.0.2-rc/mapping_00229-model.pth.tar" -O "./checkpoints/mapping_00229-model.pth.tar"
wget -nc "https://github.com/OpenTalker/SadTalker/releases/download/v0.0.2-rc/SadTalker_V0.0.2_256.safetensors" -O "./checkpoints/SadTalker_V0.0.2_256.safetensors"
wget -nc "https://github.com/OpenTalker/SadTalker/releases/download/v0.0.2-rc/SadTalker_V0.0.2_512.safetensors" -O "./checkpoints/SadTalker_V0.0.2_512.safetensors"
# Create the weights directory
mkdir -p -v "./gfpgan/weights"
# Download enhancer model files into the weights directory
wget -nc "https://github.com/xinntao/facexlib/releases/download/v0.1.0/alignment_WFLW_4HG.pth" -O "./gfpgan/weights/alignment_WFLW_4HG.pth"
wget -nc "https://github.com/xinntao/facexlib/releases/download/v0.1.0/detection_Resnet50_Final.pth" -O "./gfpgan/weights/detection_Resnet50_Final.pth"
wget -nc "https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.4.pth" -O "./gfpgan/weights/GFPGANv1.4.pth"
wget -nc "https://github.com/xinntao/facexlib/releases/download/v0.2.2/parsing_parsenet.pth" -O "./gfpgan/weights/parsing_parsenet.pth"