speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

bug fix for numpy input

Browse Source
This commit is contained in:
shixian.shi 2024-02-06 17:55:35 +08:00
parent 78ee057a99
commit 410a85402d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
funasr/models/campplus/model.py
View File

@ -123,5 +123,5 @@ class CAMPPlus(torch.nn.Module):
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