From c6574bf4f4390a9c0dfdd5aa2c4f39eb85e1557e Mon Sep 17 00:00:00 2001
From: Shi Xian <40013335+R1ckShi@users.noreply.github.com>
Date: 星期二, 09 四月 2024 10:41:11 +0800
Subject: [PATCH] Merge pull request #1596 from wanghuii1/main

---
 funasr/models/sanm/model.py |   11 ++++++++++-
 1 files changed, 10 insertions(+), 1 deletions(-)

diff --git a/funasr/models/sanm/model.py b/funasr/models/sanm/model.py
index 4dc8825..0cef540 100644
--- a/funasr/models/sanm/model.py
+++ b/funasr/models/sanm/model.py
@@ -1,3 +1,8 @@
+#!/usr/bin/env python3
+# -*- encoding: utf-8 -*-
+# Copyright FunASR (https://github.com/alibaba-damo-academy/FunASR). All Rights Reserved.
+#  MIT License  (https://opensource.org/licenses/MIT)
+
 import logging
 
 import torch
@@ -7,7 +12,11 @@
 
 @tables.register("model_classes", "SANM")
 class SANM(Transformer):
-    """CTC-attention hybrid Encoder-Decoder model"""
+    """
+    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
+    """
 
     def __init__(
         self,

--
Gitblit v1.9.1