Add C++ runtime and Python API for NeMo Canary models (#2352)

2025-07-07 17:03:49 +08:00
parent f8d957a24b
commit 0e738c356c
24 changed files with 1091 additions and 8 deletions
--- a/sherpa-onnx/csrc/onnx-utils.cc
+++ b/sherpa-onnx/csrc/onnx-utils.cc
@@ -183,6 +183,10 @@ Ort::Value View(Ort::Value *v) {
      return Ort::Value::CreateTensor(
          memory_info, v->GetTensorMutableData<float>(),
          type_and_shape.GetElementCount(), shape.data(), shape.size());
+    case ONNX_TENSOR_ELEMENT_DATA_TYPE_BOOL:
+      return Ort::Value::CreateTensor(
+          memory_info, v->GetTensorMutableData<bool>(),
+          type_and_shape.GetElementCount(), shape.data(), shape.size());
    default:
      fprintf(stderr, "Unsupported type: %d\n",
              static_cast<int32_t>(type_and_shape.GetElementType()));