From 18b1449d1ff06c469e54190508c4f6be05c73d85 Mon Sep 17 00:00:00 2001
From: 夜雨飘零 <yeyupiaoling@foxmail.com>
Date: 星期二, 05 十二月 2023 22:04:14 +0800
Subject: [PATCH] 分角色语音识别支持更多的模型

---
 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