File size: 650 Bytes
fe6336c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 | #!/usr/bin/env bash
# Create the python environment and fetch the weights from this repo.
set -e
REPO="sadmN/emotion-avatar-server"
ROOT="$(cd "$(dirname "$0")/.." && pwd)"
cd "$ROOT"
python3 -m venv .venv
source .venv/bin/activate
pip install --upgrade pip
pip install -r requirements.txt
# weights (IMTalker generator + renderer + wav2vec2 + reference image)
pip install "huggingface_hub[cli]"
huggingface-cli download "$REPO" --local-dir weights_hf \
--include "imtalker/*" "weights/wav2vec2-base-960h/*" "weights/ref.jpeg"
mkdir -p code
unzip -oq emotion_avatar_code.zip -d code
echo "done. weights in ./weights_hf, code in ./code/emo_code"
|