From eef1c2a5b885ebc045373a98cb5d3f66019c65b7 Mon Sep 17 00:00:00 2001
From: Hui Wang <86783925+wanghuii1@users.noreply.github.com>
Date: 星期二, 09 四月 2024 09:58:50 +0800
Subject: [PATCH] Update cluster_backend.py
---
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