From 5071e7b4b629338f84ca21ba0c9cdb12e68213d7 Mon Sep 17 00:00:00 2001
From: 雾聪 <wucong.lyb@alibaba-inc.com>
Date: 星期三, 20 十二月 2023 16:20:17 +0800
Subject: [PATCH] update TimestampSentence & TimestampIsPunctuation

---
 funasr/utils/whisper_utils/tokenizer.py |    7 ++++++-
 1 files changed, 6 insertions(+), 1 deletions(-)

diff --git a/funasr/utils/whisper_utils/tokenizer.py b/funasr/utils/whisper_utils/tokenizer.py
index 2a93647..bf7cb47 100644
--- a/funasr/utils/whisper_utils/tokenizer.py
+++ b/funasr/utils/whisper_utils/tokenizer.py
@@ -5,7 +5,12 @@
 
 import numpy as np
 import torch
-from transformers import GPT2TokenizerFast
+try:
+    from transformers import GPT2TokenizerFast
+except ImportError:
+    raise ImportError(
+        "transformers was not installed. Please install transformers first."
+        )
 
 LANGUAGES = {
     "en": "english",

--
Gitblit v1.9.1