funasr/modules/repeat.py
@@ -49,13 +49,12 @@ block_list: List[torch.nn.Module], output_size: int, norm_class: torch.nn.Module = torch.nn.LayerNorm, norm_args: Optional[Dict] = None, ) -> None: """Construct a MultiBlocks object.""" super().__init__() self.blocks = torch.nn.ModuleList(block_list) self.norm_blocks = norm_class(output_size, **norm_args) self.norm_blocks = norm_class(output_size) self.num_blocks = len(block_list)