funasr/torch_utils/forward_adaptor.py
@@ -1,5 +1,4 @@ import torch from typeguard import check_argument_types class ForwardAdaptor(torch.nn.Module): @@ -21,7 +20,6 @@ """ def __init__(self, module: torch.nn.Module, name: str): assert check_argument_types() super().__init__() self.module = module self.name = name