Support CED models (#792)

2024-04-19 15:20:37 +08:00
parent d97a283dbb
commit c1608b3524
33 changed files with 605 additions and 46 deletions
--- a/sherpa-onnx/csrc/offline-stream.h
+++ b/sherpa-onnx/csrc/offline-stream.h
@@ -67,13 +67,15 @@ struct OfflineFeatureExtractorConfig {
 };

 struct WhisperTag {};
+struct CEDTag {};

 class OfflineStream {
 public:
  explicit OfflineStream(const OfflineFeatureExtractorConfig &config = {},
                         ContextGraphPtr context_graph = {});

-  explicit OfflineStream(WhisperTag tag, ContextGraphPtr context_graph = {});
+  explicit OfflineStream(WhisperTag tag);
+  explicit OfflineStream(CEDTag tag);
  ~OfflineStream();

  /**