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