shixian.shi
2023-06-27 a6ccb955ed48da9c6e5bd683c0a68205a9e5f6ff
funasr/modules/repeat.py
@@ -7,7 +7,7 @@
"""Repeat the same layer definition."""
from typing import Dict, List, Optional
from funasr.modules.layer_norm import LayerNorm
import torch
@@ -48,7 +48,7 @@
        self,
        block_list: List[torch.nn.Module],
        output_size: int,
        norm_class: torch.nn.Module = torch.nn.LayerNorm,
        norm_class: torch.nn.Module = LayerNorm,
    ) -> None:
        """Construct a MultiBlocks object."""
        super().__init__()