model.py 373 B

12345678910111213141516171819
  1. import logging
  2. import torch
  3. from funasr.models.transformer.model import Transformer
  4. from funasr.register import tables
  5. @tables.register("model_classes", "Conformer")
  6. class Conformer(Transformer):
  7. """CTC-attention hybrid Encoder-Decoder model"""
  8. def __init__(
  9. self,
  10. *args,
  11. **kwargs,
  12. ):
  13. super().__init__(*args, **kwargs)