Skip to content

Commit

Permalink
Apply suggestions from code review
Browse files Browse the repository at this point in the history
Co-authored-by: Fangjun Kuang <[email protected]>
  • Loading branch information
manickavela29 and csukuangfj authored Jul 15, 2024
1 parent 9e052f5 commit 8f3b636
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
4 changes: 2 additions & 2 deletions sherpa-onnx/csrc/online-zipformer2-transducer-model.cc
Original file line number Diff line number Diff line change
Expand Up @@ -34,8 +34,8 @@ OnlineZipformer2TransducerModel::OnlineZipformer2TransducerModel(
const OnlineModelConfig &config)
: env_(ORT_LOGGING_LEVEL_WARNING),
encoder_sess_opts_(GetSessionOptions(config)),
decoder_sess_opts_(GetSessionOptions(config,"decoder")),
joiner_sess_opts_(GetSessionOptions(config,"joiner")),
decoder_sess_opts_(GetSessionOptions(config, "decoder")),
joiner_sess_opts_(GetSessionOptions(config, "joiner")),
config_(config),
allocator_{} {
{
Expand Down
2 changes: 1 addition & 1 deletion sherpa-onnx/csrc/session.cc
Original file line number Diff line number Diff line change
Expand Up @@ -231,7 +231,7 @@ Ort::SessionOptions GetSessionOptions(const OnlineModelConfig &config,
if(config.provider_config.provider == "trt" &&
(model_type == "decoder" || model_type == "joiner")) {
return GetSessionOptionsImpl(config.num_threads,
std::string("cuda"), &config.provider_config);
"cuda", &config.provider_config);
}
return GetSessionOptionsImpl(config.num_threads,
config.provider_config.provider, &config.provider_config);
Expand Down

0 comments on commit 8f3b636

Please sign in to comment.