Update rwkv_encoder.py

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

View File

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