#!/bin/bash
set -e

# Clone the YarnGPT repository
echo "Cloning YarnGPT repository..."
git clone https://github.com/saheedniyi02/yarngpt.git

# Install the package
echo "Installing YarnGPT from local source..."
cd yarngpt
pip install -e .
cd ..

# Download model files
echo "Downloading model files..."
wget -nc https://huggingface.co/novateur/WavTokenizer-medium-speech-75token/resolve/main/wavtokenizer_mediumdata_frame75_3s_nq1_code4096_dim512_kmeans200_attn.yaml
wget -nc https://huggingface.co/novateur/WavTokenizer-large-speech-75token/resolve/main/wavtokenizer_large_speech_320_24k.ckpt

echo "Installation complete!"