speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update rwkv_encoder.py

Browse Source 
delete check_argument_types
This commit is contained in:
aky15 2023-10-23 16:08:16 +08:00 committed by GitHub
parent b8d7c62b8c
commit 7ee0955bb9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
funasr/models/encoder/rwkv_encoder.py
View File

@ -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],