EngineX-Iluvatar/enginex-mr_series-sherpa-onnx
Archived
16
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

Catch exception from whisper (#408)

Browse Source
This commit is contained in:
Fangjun Kuang
2023-11-05 11:10:24 +08:00
committed by GitHub
parent d1a450bf82
commit 606cb26a62
1 changed files with 15 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
sherpa-onnx/csrc/offline-recognizer-whisper-impl.h
View File

@@ -107,8 +107,10 @@ class OfflineRecognizerWhisperImpl : public OfflineRecognizerImpl {
int32_t num_frames = f.size() / feat_dim; int32_t num_frames = f.size() / feat_dim;
if (num_frames > max_num_frames) { if (num_frames > max_num_frames) {
SHERPA_ONNX_LOGE("Only waves less than 30 seconds are supported."); SHERPA_ONNX_LOGE(
exit(-1); "Only waves less than 30 seconds are supported. We process only the "
"first 30 seconds and discard the remaining data");
num_frames = max_num_frames;
} }
NormalizeFeatures(f.data(), num_frames, feat_dim); NormalizeFeatures(f.data(), num_frames, feat_dim);
@@ -124,13 +126,19 @@ class OfflineRecognizerWhisperImpl : public OfflineRecognizerImpl {
(max_num_frames - num_frames) * feat_dim * sizeof(float)); (max_num_frames - num_frames) * feat_dim * sizeof(float));
mel = Transpose12(model_->Allocator(), &mel); mel = Transpose12(model_->Allocator(), &mel);
auto cross_kv = model_->ForwardEncoder(std::move(mel)); try {
auto cross_kv = model_->ForwardEncoder(std::move(mel));
auto results = auto results = decoder_->Decode(std::move(cross_kv.first),
decoder_->Decode(std::move(cross_kv.first), std::move(cross_kv.second)); std::move(cross_kv.second));
auto r = Convert(results[0], symbol_table_); auto r = Convert(results[0], symbol_table_);
s->SetResult(r); s->SetResult(r);
} catch (const Ort::Exception &ex) {
SHERPA_ONNX_LOGE("\n\nCaught exception:\n\n%s\n\nReturn an empty result",
ex.what());
return;
}
} }
private: private: