#!/usr/bin/env bash
# F5-TTS 전용 venv (NVIDIA CUDA)
set -euo pipefail

ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
VENV="$ROOT/.venvs/f5tts"

python3 -m venv "$VENV"
"$VENV/bin/pip" install -U pip wheel

# CUDA 12.x PyTorch (서버 CUDA 버전에 맞게 cu124/cu128 조정)
"$VENV/bin/pip" install torch torchaudio --index-url https://download.pytorch.org/whl/cu124
"$VENV/bin/pip" install f5-tts

echo "F5-TTS venv 준비 완료: $VENV"
echo "테스트: $VENV/bin/python $ROOT/scripts/workers/f5_infer.py --help"