Ver Fonte

Update rwkv_encoder.py

delete check_argument_types
aky15 há 2 anos atrás
pai
commit
7ee0955bb9
1 ficheiros alterados com 0 adições e 3 exclusões
  1. 0 3
      funasr/models/encoder/rwkv_encoder.py

+ 0 - 3
funasr/models/encoder/rwkv_encoder.py

@@ -4,7 +4,6 @@ import math
 from typing import Dict, List, Optional, Tuple
 from typing import Dict, List, Optional, Tuple
 
 
 import torch
 import torch
-from typeguard import check_argument_types
 
 
 from funasr.models.encoder.abs_encoder import AbsEncoder
 from funasr.models.encoder.abs_encoder import AbsEncoder
 from funasr.modules.rwkv import RWKV
 from funasr.modules.rwkv import RWKV
@@ -49,8 +48,6 @@ class RWKVEncoder(AbsEncoder):
         """Construct a RWKVEncoder object."""
         """Construct a RWKVEncoder object."""
         super().__init__()
         super().__init__()
 
 
-        assert check_argument_types()
-
         self.embed = RWKVConvInput(
         self.embed = RWKVConvInput(
             input_size,
             input_size,
             [output_size//4, output_size//2, output_size],
             [output_size//4, output_size//2, output_size],