From 70645e48072bf193fbf069949f1d2b10fddac8a3 Mon Sep 17 00:00:00 2001
From: pointerhacker <145901472+pointerhacker@users.noreply.github.com>
Date: 星期二, 15 十月 2024 17:50:51 +0800
Subject: [PATCH] 数据并行可能导致的模型训练报错 (#2139)
---
runtime/python/onnxruntime/setup.py | 5 +++--
1 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/runtime/python/onnxruntime/setup.py b/runtime/python/onnxruntime/setup.py
index b2977ce..26dce2c 100644
--- a/runtime/python/onnxruntime/setup.py
+++ b/runtime/python/onnxruntime/setup.py
@@ -13,7 +13,7 @@
MODULE_NAME = "funasr_onnx"
-VERSION_NUM = "0.3.1"
+VERSION_NUM = "0.4.1"
setuptools.setup(
name=MODULE_NAME,
@@ -31,10 +31,11 @@
"librosa",
"onnxruntime>=1.7.0",
"scipy",
- "numpy>=1.19.3",
+ "numpy<=1.26.4",
"kaldi-native-fbank",
"PyYAML>=5.1.2",
"onnx",
+ "sentencepiece",
],
packages=[MODULE_NAME, f"{MODULE_NAME}.utils"],
keywords=["funasr,asr"],
--
Gitblit v1.9.1