游雁
2024-02-19 94de39dde2e616a01683c518023d0fab72b4e103
funasr/models/branchformer/model.py
@@ -1,16 +1,16 @@
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)