marcoyang's picture
add wav and inference script
60c8de6
raw
history blame contribute delete
578 Bytes
#!/usr/bin/env bash
source ~/softwares/pyenvs/k2_cuda11/k2_cuda11/bin/activate
export PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python
export PYTHONPATH=/star-xy/softwares/icefall_development/icefall_multi_KD_streaming:$PYTHONPATH
export PYTHONPATH=/star-xy/softwares/whisper:$PYTHONPATH
export CUDA_VISIBLE_DEVICES="0"
python multi_KD_inference/inference_model.py \
--speaker-input-idx 2 \
--whisper-dim 1280 \
--causal 0 \
--ckpt-path icefall-multi-kd-zipformer/pretrained.pt \
--extract-spkr-embed False \
--audio icefall-multi-kd-zipformer/00001.wav