From cec4a56dab599aae2a94ff79fdc655276bd44545 Mon Sep 17 00:00:00 2001
From: 志浩 <neo.dzh@alibaba-inc.com>
Date: 星期四, 15 六月 2023 19:12:13 +0800
Subject: [PATCH] modify sv_model_version for speech_diarization_sond-zh-cn-alimeeting-16k-n16k4-pytorch/infer.py

---
 funasr/runtime/triton_gpu/model_repo_paraformer_large_offline/encoder/config.pbtxt |    2 +-
 1 files changed, 1 insertions(+), 1 deletions(-)

diff --git a/funasr/runtime/triton_gpu/model_repo_paraformer_large_offline/encoder/config.pbtxt b/funasr/runtime/triton_gpu/model_repo_paraformer_large_offline/encoder/config.pbtxt
index 8d89993..3cb657e 100644
--- a/funasr/runtime/triton_gpu/model_repo_paraformer_large_offline/encoder/config.pbtxt
+++ b/funasr/runtime/triton_gpu/model_repo_paraformer_large_offline/encoder/config.pbtxt
@@ -40,7 +40,7 @@
   },
   {
     name: "token_num"
-    data_type: TYPE_INT64
+    data_type: TYPE_INT32
     dims: [1]
     reshape: { shape: [ ] }
   }

--
Gitblit v1.9.1