run_180m_flash.sh
2.1 KB
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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
#!/usr/bin/env bash
# Copyright 2025 Xiaomi Corp. (authors: Fangjun Kuang)
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]}) $*"
}
curl -SL -O https://github.com/k2-fsa/sherpa-onnx/releases/download/asr-models/de.wav
curl -SL -O https://github.com/k2-fsa/sherpa-onnx/releases/download/asr-models/en.wav
pip install \
nemo_toolkit['asr'] \
"numpy<2" \
ipython \
kaldi-native-fbank \
librosa \
onnx==1.17.0 \
onnxruntime==1.17.1 \
onnxscript \
soundfile
python3 ./export_onnx_180m_flash.py
ls -lh *.onnx
log "-----fp32------"
python3 ./test_180m_flash.py \
--encoder ./encoder.onnx \
--decoder ./decoder.onnx \
--source-lang en \
--target-lang en \
--tokens ./tokens.txt \
--wav ./en.wav
python3 ./test_180m_flash.py \
--encoder ./encoder.onnx \
--decoder ./decoder.onnx \
--source-lang en \
--target-lang de \
--tokens ./tokens.txt \
--wav ./en.wav
python3 ./test_180m_flash.py \
--encoder ./encoder.onnx \
--decoder ./decoder.onnx \
--source-lang de \
--target-lang de \
--tokens ./tokens.txt \
--wav ./de.wav
python3 ./test_180m_flash.py \
--encoder ./encoder.onnx \
--decoder ./decoder.onnx \
--source-lang de \
--target-lang en \
--tokens ./tokens.txt \
--wav ./de.wav
log "-----int8------"
python3 ./test_180m_flash.py \
--encoder ./encoder.int8.onnx \
--decoder ./decoder.int8.onnx \
--source-lang en \
--target-lang en \
--tokens ./tokens.txt \
--wav ./en.wav
python3 ./test_180m_flash.py \
--encoder ./encoder.int8.onnx \
--decoder ./decoder.int8.onnx \
--source-lang en \
--target-lang de \
--tokens ./tokens.txt \
--wav ./en.wav
python3 ./test_180m_flash.py \
--encoder ./encoder.int8.onnx \
--decoder ./decoder.int8.onnx \
--source-lang de \
--target-lang de \
--tokens ./tokens.txt \
--wav ./de.wav
python3 ./test_180m_flash.py \
--encoder ./encoder.int8.onnx \
--decoder ./decoder.int8.onnx \
--source-lang de \
--target-lang en \
--tokens ./tokens.txt \
--wav ./de.wav