From 7ee0955bb91a81ffc1c911eddba4bd5283625fc2 Mon Sep 17 00:00:00 2001 From: aky15 Date: Mon, 23 Oct 2023 16:08:16 +0800 Subject: [PATCH] Update rwkv_encoder.py delete check_argument_types --- funasr/models/encoder/rwkv_encoder.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/funasr/models/encoder/rwkv_encoder.py b/funasr/models/encoder/rwkv_encoder.py index 291ed1922..8a33520e9 100644 --- a/funasr/models/encoder/rwkv_encoder.py +++ b/funasr/models/encoder/rwkv_encoder.py @@ -4,7 +4,6 @@ import math from typing import Dict, List, Optional, Tuple import torch -from typeguard import check_argument_types from funasr.models.encoder.abs_encoder import AbsEncoder from funasr.modules.rwkv import RWKV @@ -49,8 +48,6 @@ class RWKVEncoder(AbsEncoder): """Construct a RWKVEncoder object.""" super().__init__() - assert check_argument_types() - self.embed = RWKVConvInput( input_size, [output_size//4, output_size//2, output_size],