update funasr-onnx-offline-rtf

雾聪

2023-06-02 8b6a1103366f3ccdaa4e75aa663094a3570c6ae3

 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,14 +48,13 @@
        self,
        block_list: List[torch.nn.Module],
        output_size: int,
        norm_class: torch.nn.Module = torch.nn.LayerNorm,
        norm_args: Optional[Dict] = None,
        norm_class: torch.nn.Module = LayerNorm,
    ) -> 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)