File size: 2,494 Bytes
6e802a4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 | #!/bin/bash
# Decode a prompt list with one model using the report's settings (decode H, voice 4200, seed 1234, cached).
# bash decode.sh <model_dir> <prompts.jsonl> <out_dir> [selfplan|noplan]
# env: PYTHON, SPK (4200 | row), SEED, ROWS=a:b, REPAIR=0, FISH_PYTHON (+ FISH_SPEECH_REPO, FISH_CODEC_PTH) for wavs
set -euo pipefail
HERE=$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)
[ $# -ge 3 ] || { sed -n '2,/^set -e/p' "$0" | grep '^#'; exit 1; }
MODEL=$(cd "$1" && pwd)
PS=$(readlink -f "$2")
mkdir -p "$3"; OUT=$(cd "$3" && pwd)
CELL=${4:-selfplan}
PY=${PYTHON:-python}
SPK=${SPK:-4200}; SEED=${SEED:-1234}; REPAIR=${REPAIR:-1}
export MVC_NUM_SPEECH_TOKENS=2048 MVC_NUM_CODEC_LEVELS=10 MVC_ENABLE_PLAN_TOKENS=1
[ -f "$MODEL/model.safetensors" ] || { echo "no model.safetensors in $MODEL"; exit 1; }
FLAGS=$("$PY" -c 'import json,sys; print(" ".join(json.load(open(sys.argv[1]))["generator_flags"]))' "$MODEL/config.json")
DEC="--temp_semantic 0.7 --acoustic_temps 0.80 0.7 0.7 0.7 0.7 0.7 0.7 0.7 0.7 \
--rep_penalty_levels 1.2 1.6 1.2 1.2 1.2 1.2 1.2 1.2 1.2 1.2 \
--ras_tau 0.1 --ras_win 10 --top_p 0.8 --rep_win 16 --stop_threshold 0.02"
if [ "$REPAIR" = "1" ]; then
DEC="$DEC --loop_break_flipk 5.0 --loop_break_win 32 --loop_break_temp 0.6 --loop_break_rep 2.0 \
--loop_break_stop_after 24 --silence_guard_codes $HERE/data/silence_codes_fish_s2.json \
--silence_guard_budget 10 --silence_guard_stop_after 10 --silence_guard_retries 3 \
--silence_guard_retry_temp 0.6 --silence_guard_stop_noplan"
fi
DEC="$DEC --cached"
case "$CELL" in
selfplan) MODE="--mode self --cot_mode plan --plan_window --plan_dur_lo 0.8 --plan_dur_hi 1.5 \
--plan_speech_ratio 0.660 --plan_injection" ;;
noplan) MODE="--mode nocot --duration_model $HERE/data/duration_model.json --dur_lo 0.9 --dur_hi 1.5" ;;
*) echo "cell must be selfplan or noplan"; exit 1 ;;
esac
if [ -n "${ROWS:-}" ]; then
SEL="--rows $(seq -s ' ' "${ROWS%%:*}" $(( ${ROWS##*:} - 1 )))"
else
SEL="--limit $(grep -c . "$PS")"
fi
if [ "$SPK" = "row" ]; then SPKARG=""; else SPKARG="--force_speaker_id $SPK"; fi
cd "$HERE"
"$PY" scripts/gen_fish_cot_samples.py --val "$PS" $SEL --dtype fp32 $MODE \
--checkpoint "$MODEL/model.safetensors" $SPKARG --seed "$SEED" --max_think 0 \
$FLAGS $DEC --out "$OUT/$CELL.jsonl"
if [ -n "${FISH_PYTHON:-}" ]; then
"$FISH_PYTHON" scripts/decode_fish_codes.py --codes "$OUT/$CELL.jsonl" --outdir "$OUT/$CELL"
fi
|