From a8591060d3889cd7a72841fa32a7ee64b49db1d2 Mon Sep 17 00:00:00 2001 From: msgk <zxr935867802@outlook.com> Date: 星期五, 14 二月 2025 14:16:51 +0800 Subject: [PATCH] fix(spk): 修复 speaker embedding 集群后的重新排序问题 --- export.py | 5 +++-- 1 files changed, 3 insertions(+), 2 deletions(-) diff --git a/export.py b/export.py index 3307e96..a891e60 100644 --- a/export.py +++ b/export.py @@ -2,8 +2,9 @@ from funasr import AutoModel model = AutoModel( - model="/raid/t3cv/wangch/WORK_SAPCE/ASR/models/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch" + model="iic/emotion2vec_base", + hub="ms" ) -res = model.export(type="onnx", quantize=False, opset_version=13, device='cuda') # fp32 onnx-gpu +res = model.export(type="onnx", quantize=False, opset_version=13, device='cpu') # fp32 onnx-gpu # res = model.export(type="onnx_fp16", quantize=False, opset_version=13, device='cuda') # fp16 onnx-gpu -- Gitblit v1.9.1