bug fix for numpy input

modelscope · Feb 6, 2024 · 410a854 · 410a854
1 parent 78ee057
commit 410a854
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/funasr/models/campplus/model.py b/funasr/models/campplus/model.py
@@ -123,5 +123,5 @@ def inference(self,
         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