nanobot-voice-interface/scripts/install_melotts_cpu.sh

52 lines
1.3 KiB
Bash

#!/usr/bin/env bash
set -euo pipefail
ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
VENV_PYTHON="${ROOT_DIR}/.venv/bin/python"
if [[ ! -x "${VENV_PYTHON}" ]]; then
echo "error: ${VENV_PYTHON} does not exist. Create the web UI virtualenv first." >&2
exit 1
fi
"${VENV_PYTHON}" -m pip install \
--index-url https://download.pytorch.org/whl/cpu \
"torch==2.7.1+cpu" \
"torchaudio==2.7.1+cpu"
"${VENV_PYTHON}" -m pip install "setuptools<81"
"${VENV_PYTHON}" -m pip install \
txtsplit \
cached_path \
"transformers==4.46.3" \
"num2words==0.5.12" \
"unidic_lite==1.0.8" \
"mecab-python3==1.0.9" \
fugashi \
"pykakasi==2.2.1" \
"g2p_en==2.1.0" \
"anyascii==0.3.2" \
"jamo==0.4.1" \
"gruut[de,es,fr]==2.2.3" \
"librosa==0.9.1" \
"pydub==0.25.1" \
"eng_to_ipa==0.0.2" \
"inflect==7.0.0" \
"unidecode==1.3.7" \
"pypinyin==0.50.0" \
"cn2an==0.5.22" \
"jieba==0.42.1" \
soundfile \
tqdm
"${VENV_PYTHON}" -m pip install --no-deps "git+https://github.com/myshell-ai/MeloTTS.git"
"${VENV_PYTHON}" - <<'PY'
import os
import nltk
download_dir = os.path.expanduser("~/nltk_data")
for package in ("averaged_perceptron_tagger", "averaged_perceptron_tagger_eng", "cmudict"):
nltk.download(package, download_dir=download_dir)
PY