87 lines
1.8 KiB
Bash
87 lines
1.8 KiB
Bash
#!/usr/bin/env bash
|
|
# Copyright 2025 Xiaomi Corp. (authors: Fangjun Kuang)
|
|
#
|
|
# Auto generated! Do NOT edit!
|
|
|
|
set -ex
|
|
|
|
log() {
|
|
# This function is from espnet
|
|
local fname=${BASH_SOURCE[1]##*/}
|
|
echo -e "$(date '+%Y-%m-%d %H:%M:%S') (${fname}:${BASH_LINENO[0]}:${FUNCNAME[1]}) $*"
|
|
}
|
|
|
|
wget -qq https://github.com/k2-fsa/sherpa-onnx/releases/download/tts-models/espeak-ng-data.tar.bz2
|
|
tar xf espeak-ng-data.tar.bz2
|
|
rm espeak-ng-data.tar.bz2
|
|
|
|
mkdir -p release
|
|
|
|
{% for model in model_list %}
|
|
|
|
name={{ model.name }}
|
|
kind={{ model.kind }}
|
|
lang={{ model.lang }}
|
|
model_name={{ model.model_name }}
|
|
text="{{ model.text }}"
|
|
num_speakers={{ model.ns }}
|
|
sample_rate={{ model.sr }}
|
|
|
|
{{ model.cmd }}
|
|
|
|
python3 ./add_meta_data.py \
|
|
--name $name \
|
|
--kind $kind \
|
|
--lang $lang
|
|
|
|
dst=vits-piper-$lang-$name-$kind
|
|
dst_int8=vits-piper-$lang-$name-$kind-int8
|
|
dst_fp16=vits-piper-$lang-$name-$kind-fp16
|
|
mkdir -p $dst
|
|
|
|
mv -v tokens.txt $dst/
|
|
mv -v MODEL_CARD $dst/ || true
|
|
mv -v README $dst/ || true
|
|
mv -v *.json $dst/
|
|
cp -a ./espeak-ng-data $dst/
|
|
|
|
cp -a $dst $dst_int8
|
|
cp -a $dst $dst_fp16
|
|
|
|
mv -v *.onnx $dst/
|
|
|
|
python3 ./dynamic_quantization.py \
|
|
--input $dst/$model_name \
|
|
--output-int8 $dst_int8/$model_name \
|
|
--output-fp16 $dst_fp16/$model_name >/dev/null 2>&1
|
|
|
|
echo "---fp32---"
|
|
ls -lh $dst
|
|
|
|
echo "---int8---"
|
|
ls -lh $dst_int8
|
|
|
|
echo "---fp16---"
|
|
ls -lh $dst_fp16
|
|
|
|
tar cjf ${dst}.tar.bz2 $dst
|
|
tar cjf ${dst_int8}.tar.bz2 $dst_int8
|
|
tar cjf ${dst_fp16}.tar.bz2 $dst_fp16
|
|
|
|
if [ -d hf ]; then
|
|
mkdir -p hf/piper/mp3/$lang/vits-piper-$lang-$name-$kind
|
|
for i in $(seq $num_speakers); do
|
|
i=$((i-1))
|
|
python3 ./generate_samples-$dst-$i.py
|
|
done
|
|
ls -lh hf/piper/mp3/$lang/vits-piper-$lang-$name-$kind
|
|
fi
|
|
|
|
mv $dst release
|
|
mv $dst_int8 release
|
|
mv $dst_fp16 release
|
|
|
|
ls -lh release/*
|
|
|
|
{% endfor %}
|