mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-08 09:32:20 +00:00
16a2748d6c
* Add PromptASR with BERT as text encoder * Support using word-list based content prompts for context biasing * Upload the pretrained models to huggingface * Add usage example
37 lines
919 B
Bash
Executable File
37 lines
919 B
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
set -eou pipefail
|
|
|
|
# This is the preparation recipe for PromptASR: https://arxiv.org/pdf/2309.07414
|
|
|
|
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]}) $*"
|
|
}
|
|
|
|
stage=-1
|
|
stop_stage=100
|
|
manifest_dir=data/fbank
|
|
subset=medium
|
|
topk=10000
|
|
|
|
. shared/parse_options.sh || exit 1
|
|
|
|
if [ $stage -le 0 ] && [ $stop_stage -ge 0 ]; then
|
|
log "Stage 0: Download the meta biasing list for LibriSpeech"
|
|
mkdir -p data/context_biasing
|
|
cd data/context_biasing
|
|
git clone https://github.com/facebookresearch/fbai-speech.git
|
|
cd ../..
|
|
fi
|
|
|
|
if [ $stage -le 1 ] && [ $stop_stage -ge 1 ]; then
|
|
log "Stage 1: Add rare-words for context biasing to the manifest"
|
|
python zipformer_prompt_asr/utils.py \
|
|
--manifest-dir $manifest_dir \
|
|
--subset $subset \
|
|
--top-k $topk
|
|
|
|
fi
|