From 997374b88fe6b2ae5cb4dcaf47d78cb3eff09fc2 Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期二, 11 六月 2024 19:56:52 +0800
Subject: [PATCH] add ctc inference code (#1806)
---
funasr/models/ctc/ctc.py | 4 +---
1 files changed, 1 insertions(+), 3 deletions(-)
diff --git a/funasr/models/ctc/ctc.py b/funasr/models/ctc/ctc.py
index d3c10fa..bdfb3a6 100644
--- a/funasr/models/ctc/ctc.py
+++ b/funasr/models/ctc/ctc.py
@@ -40,9 +40,7 @@
logging.warning("ignore_nan_grad option is not supported for warp_ctc")
self.ctc_loss = warp_ctc.CTCLoss(size_average=True, reduce=reduce)
else:
- raise ValueError(
- f'ctc_type must be "builtin" or "warpctc": {self.ctc_type}'
- )
+ raise ValueError(f'ctc_type must be "builtin" or "warpctc": {self.ctc_type}')
self.reduce = reduce
--
Gitblit v1.9.1