From 9a9b474e7de7cc90d2ee124dc8d6c2cfa887c059 Mon Sep 17 00:00:00 2001
From: xiaowan0322 <wanchen.swc@alibaba-inc.com>
Date: 星期四, 06 六月 2024 15:59:56 +0800
Subject: [PATCH] [Optimization] support bladedisc fp16 optimization (#1790)
---
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