#!/bin/bash

# Since we use preloaded, cached versions of models, we must link them for further use
mkdir ~/app/models/musetalkV15
ln -s ~/.cache/huggingface/hub/models--TMElyralab--MuseTalk/snapshots/*/musetalkV15/* ~/app/models/musetalkV15/

mkdir ~/app/models/musetalk
ln -s ~/.cache/huggingface/hub/models--TMElyralab--MuseTalk/snapshots/*/musetalk/* ~/app/models/musetalk/

mkdir ~/app/models/dwpose
ln -s ~/.cache/huggingface/hub/models--yzd-v--DWPose/snapshots/*/* ~/app/models/dwpose/

mkdir ~/app/models/sd-vae
ln -s ~/.cache/huggingface/hub/models--stabilityai--sd-vae-ft-mse/snapshots/*/* ~/app/models/sd-vae/

mkdir ~/app/models/syncnet
ln -s ~/.cache/huggingface/hub/models--ByteDance--LatentSync/snapshots/*/* ~/app/models/syncnet/

mkdir ~/app/models/whisper
ln -s ~/.cache/huggingface/hub/models--openai--whisper-tiny/snapshots/*/* ~/app/models/whisper/

pip install --no-cache-dir -U openmim

# Mim uses ~/.cache/mim by default for cache, but hugging face only allows to use /home/user/app, so we use little hack
export HOME=/home/user/app

mim install mmengine
mim install "mmcv==2.1.0"
mim install "mmdet==3.2.0"
mim install "mmpose==1.2.0"