正在显示
1 个修改的文件
包含
4 行增加
和
0 行删除
| @@ -22,18 +22,22 @@ | @@ -22,18 +22,22 @@ | ||
| 22 | namespace sherpa_onnx { | 22 | namespace sherpa_onnx { |
| 23 | 23 | ||
| 24 | std::unique_ptr<VadModel> VadModel::Create(const VadModelConfig &config) { | 24 | std::unique_ptr<VadModel> VadModel::Create(const VadModelConfig &config) { |
| 25 | +#if SHERPA_ONNX_ENABLE_RKNN | ||
| 25 | if (config.provider == "rknn") { | 26 | if (config.provider == "rknn") { |
| 26 | return std::make_unique<SileroVadModelRknn>(config); | 27 | return std::make_unique<SileroVadModelRknn>(config); |
| 27 | } | 28 | } |
| 29 | +#endif | ||
| 28 | return std::make_unique<SileroVadModel>(config); | 30 | return std::make_unique<SileroVadModel>(config); |
| 29 | } | 31 | } |
| 30 | 32 | ||
| 31 | template <typename Manager> | 33 | template <typename Manager> |
| 32 | std::unique_ptr<VadModel> VadModel::Create(Manager *mgr, | 34 | std::unique_ptr<VadModel> VadModel::Create(Manager *mgr, |
| 33 | const VadModelConfig &config) { | 35 | const VadModelConfig &config) { |
| 36 | +#if SHERPA_ONNX_ENABLE_RKNN | ||
| 34 | if (config.provider == "rknn") { | 37 | if (config.provider == "rknn") { |
| 35 | return std::make_unique<SileroVadModelRknn>(mgr, config); | 38 | return std::make_unique<SileroVadModelRknn>(mgr, config); |
| 36 | } | 39 | } |
| 40 | +#endif | ||
| 37 | return std::make_unique<SileroVadModel>(mgr, config); | 41 | return std::make_unique<SileroVadModel>(mgr, config); |
| 38 | } | 42 | } |
| 39 | 43 |
-
请 注册 或 登录 后发表评论