o
    ´©iµ  ã                   @   sH   d dl Z d dlZd dlmZ d dlmZ e dd¡G dd„ deƒƒZdS )é    N)ÚTransformer)ÚtablesÚmodel_classesÚSANMc                       s    e Zd ZdZ‡ fdd„Z‡  ZS )r   z¶
    Author: Zhifu Gao, Shiliang Zhang, Ming Lei, Ian McLoughlin
    San-m: Memory equipped self-attention for end-to-end speech recognition
    https://arxiv.org/abs/2006.01713
    c                    s   t ƒ j|i |¤Ž d S )N)ÚsuperÚ__init__)ÚselfÚargsÚkwargs©Ú	__class__© úL/home/ubuntu/.local/lib/python3.10/site-packages/funasr/models/sanm/model.pyr      s   zSANM.__init__)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   Ú__classcell__r   r   r   r   r      s    )ÚloggingÚtorchÚfunasr.models.transformer.modelr   Úfunasr.registerr   Úregisterr   r   r   r   r   Ú<module>   s   
