Support whisper models (#238)

2023-08-07 12:34:18 +08:00
parent 64efbd82af
commit 45b9d4ab37
39 changed files with 1836 additions and 52 deletions
--- a/sherpa-onnx/csrc/offline-stream.h
+++ b/sherpa-onnx/csrc/offline-stream.h
@@ -65,10 +65,15 @@ struct OfflineFeatureExtractorConfig {
  void Register(ParseOptions *po);
 };

+struct WhisperTag {};
+
 class OfflineStream {
 public:
  explicit OfflineStream(const OfflineFeatureExtractorConfig &config = {},
                         ContextGraphPtr context_graph = nullptr);
+
+  explicit OfflineStream(WhisperTag tag,
+                         ContextGraphPtr context_graph = nullptr);
  ~OfflineStream();

  /**