From b4affb21ef34ac11cb0252b386080e6494fbfb30 Mon Sep 17 00:00:00 2001
From: xuan <admin@exuan.org>
Date: 星期一, 26 二月 2024 09:41:22 +0800
Subject: [PATCH] fix: TabError (#1388)
---
funasr/models/campplus/model.py | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/funasr/models/campplus/model.py b/funasr/models/campplus/model.py
index 6706c84..b2d65f1 100644
--- a/funasr/models/campplus/model.py
+++ b/funasr/models/campplus/model.py
@@ -123,5 +123,5 @@
time3 = time.perf_counter()
meta_data["extract_feat"] = f"{time3 - time2:0.3f}"
meta_data["batch_data_time"] = np.array(speech_times).sum().item() / 16000.0
- results = [{"spk_embedding": self.forward(speech)}]
+ results = [{"spk_embedding": self.forward(speech.to(torch.float32))}]
return results, meta_data
\ No newline at end of file
--
Gitblit v1.9.1