From c644ac8f58895b9e29e9cfca79465fd2c0efaa5a Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期二, 21 十一月 2023 14:09:01 +0800
Subject: [PATCH] funasr v2 setup

---
 funasr/models/frontend/default.py |    5 ++++-
 1 files changed, 4 insertions(+), 1 deletions(-)

diff --git a/funasr/models/frontend/default.py b/funasr/models/frontend/default.py
index b41af80..8d60e20 100644
--- a/funasr/models/frontend/default.py
+++ b/funasr/models/frontend/default.py
@@ -6,7 +6,10 @@
 import humanfriendly
 import numpy as np
 import torch
-from torch_complex.tensor import ComplexTensor
+try:
+    from torch_complex.tensor import ComplexTensor
+except:
+    raise "Please install torch_complex firstly"
 
 from funasr.layers.log_mel import LogMel
 from funasr.layers.stft import Stft

--
Gitblit v1.9.1