#!/bin/bash

echo "Downloading OmniAvatar-14B models..."

# Create directories
mkdir -p pretrained_models

# Install huggingface-hub if not already installed
pip install "huggingface_hub[cli]"

# Download models
echo "Downloading Wan2.1-T2V-14B..."
huggingface-cli download Wan-AI/Wan2.1-T2V-14B --local-dir ./pretrained_models/Wan2.1-T2V-14B

echo "Downloading wav2vec2-base-960h..."
huggingface-cli download facebook/wav2vec2-base-960h --local-dir ./pretrained_models/wav2vec2-base-960h

echo "Downloading OmniAvatar-14B..."
huggingface-cli download OmniAvatar/OmniAvatar-14B --local-dir ./pretrained_models/OmniAvatar-14B

echo "Model download completed!"