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 |    8 ++++----
 1 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/funasr/models/frontend/default.py b/funasr/models/frontend/default.py
index abbcd1b..8d60e20 100644
--- a/funasr/models/frontend/default.py
+++ b/funasr/models/frontend/default.py
@@ -6,8 +6,10 @@
 import humanfriendly
 import numpy as np
 import torch
-from torch_complex.tensor import ComplexTensor
-from typeguard import check_argument_types
+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
@@ -40,7 +42,6 @@
             apply_stft: bool = True,
             use_channel: int = None,
     ):
-        assert check_argument_types()
         super().__init__()
         if isinstance(fs, str):
             fs = humanfriendly.parse_size(fs)
@@ -167,7 +168,6 @@
             cmvn_file: str = None,
             mc: bool = True
     ):
-        assert check_argument_types()
         super().__init__()
         if isinstance(fs, str):
             fs = humanfriendly.parse_size(fs)

--
Gitblit v1.9.1