From a5b5e64fdc1ea7f3cff03b3ce96015bcc3ed7828 Mon Sep 17 00:00:00 2001
From: hnluo <haoneng.lhn@alibaba-inc.com>
Date: 星期一, 13 三月 2023 11:25:12 +0800
Subject: [PATCH] Update test_asr_inference_pipeline.py
---
funasr/models/e2e_asr_paraformer.py | 1 +
1 files changed, 1 insertions(+), 0 deletions(-)
diff --git a/funasr/models/e2e_asr_paraformer.py b/funasr/models/e2e_asr_paraformer.py
index 5786bc4..8439f40 100644
--- a/funasr/models/e2e_asr_paraformer.py
+++ b/funasr/models/e2e_asr_paraformer.py
@@ -978,6 +978,7 @@
loss, stats, weight = force_gatherable((loss, stats, batch_size), loss.device)
return loss, stats, weight
+
class ContextualParaformer(Paraformer):
"""
Paraformer model with contextual hotword
--
Gitblit v1.9.1