From fb176404cfeb40c053f4f42d01eb45c185d21ce2 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期一, 08 一月 2024 16:20:45 +0800
Subject: [PATCH] funasr1.0 emotion2vec

---
 funasr/bin/inference.py |    3 ++-
 1 files changed, 2 insertions(+), 1 deletions(-)

diff --git a/funasr/bin/inference.py b/funasr/bin/inference.py
index 1676c30..5b58907 100644
--- a/funasr/bin/inference.py
+++ b/funasr/bin/inference.py
@@ -222,7 +222,8 @@
 				batch["data_lengths"] = input_len
 		
 			time1 = time.perf_counter()
-			results, meta_data = model.generate(**batch, **kwargs)
+			with torch.no_grad():
+				results, meta_data = model.generate(**batch, **kwargs)
 			time2 = time.perf_counter()
 			
 			asr_result_list.extend(results)

--
Gitblit v1.9.1