mirror of
https://github.com/modelscope/FunASR
synced 2025-09-15 14:48:36 +08:00
Update rwkv_encoder.py
delete check_argument_types
This commit is contained in:
parent
b8d7c62b8c
commit
7ee0955bb9
@ -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],
|
||||
|
||||
Loading…
Reference in New Issue
Block a user