wuhongsheng
2024-07-05 3a4281f4959534b1bf5d01acf0085f4f8e6f2ec8
funasr/models/branchformer/model.py
@@ -1,16 +1,17 @@
import logging
from funasr.models.transformer.model import Transformer
from funasr.utils.register import register_class
from funasr.register import tables
@register_class("model_classes", "Branchformer")
@tables.register("model_classes", "Branchformer")
class Branchformer(Transformer):
   """CTC-attention hybrid Encoder-Decoder model"""
    """CTC-attention hybrid Encoder-Decoder model"""
   def __init__(
      self,
      *args,
      **kwargs,
   ):
    def __init__(
        self,
        *args,
        **kwargs,
    ):
      super().__init__(*args, **kwargs)
        super().__init__(*args, **kwargs)