From c29e5071bdb0dea46a7778f3a8a08692458f19f1 Mon Sep 17 00:00:00 2001
From: Marlowe <54339989+ZihanLiao@users.noreply.github.com>
Date: 星期五, 01 九月 2023 01:15:32 +0800
Subject: [PATCH] fix error:reference before assignment for write_flag when using kaldi_ark (#904)

---
 funasr/models/target_delay_transformer.py |    6 +++++-
 1 files changed, 5 insertions(+), 1 deletions(-)

diff --git a/funasr/models/target_delay_transformer.py b/funasr/models/target_delay_transformer.py
index 683b804..19e5c7c 100644
--- a/funasr/models/target_delay_transformer.py
+++ b/funasr/models/target_delay_transformer.py
@@ -10,7 +10,11 @@
 
 
 class TargetDelayTransformer(torch.nn.Module):
-
+    """
+    Author: Speech Lab of DAMO Academy, Alibaba Group
+    CT-Transformer: Controllable time-delay transformer for real-time punctuation prediction and disfluency detection
+    https://arxiv.org/pdf/2003.01309.pdf
+    """
     def __init__(
         self,
         vocab_size: int,

--
Gitblit v1.9.1