Przeglądaj źródła

Update rwkv_encoder.py

delete check_argument_types
aky15 2 lat temu
rodzic
commit
7ee0955bb9
1 zmienionych plików z 0 dodań i 3 usunięć
  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
 
 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],