From 6427c834dfd97b1f05c6659cdc7ccf010bf82fe1 Mon Sep 17 00:00:00 2001
From: 嘉渊 <wangjiaming.wjm@alibaba-inc.com>
Date: 星期一, 24 四月 2023 19:50:07 +0800
Subject: [PATCH] update
---
funasr/runtime/python/libtorch/setup.py | 12 ++++++------
1 files changed, 6 insertions(+), 6 deletions(-)
diff --git a/funasr/runtime/python/libtorch/setup.py b/funasr/runtime/python/libtorch/setup.py
index 99d8b52..1560950 100644
--- a/funasr/runtime/python/libtorch/setup.py
+++ b/funasr/runtime/python/libtorch/setup.py
@@ -1,7 +1,7 @@
# -*- encoding: utf-8 -*-
from pathlib import Path
import setuptools
-
+from setuptools import find_packages
def get_readme():
root_dir = Path(__file__).resolve().parent
@@ -14,8 +14,8 @@
setuptools.setup(
- name='torch_paraformer',
- version='0.0.1',
+ name='funasr_torch',
+ version='0.0.3',
platforms="Any",
url="https://github.com/alibaba-damo-academy/FunASR.git",
author="Speech Lab, Alibaba Group, China",
@@ -28,10 +28,10 @@
install_requires=["librosa", "onnxruntime>=1.7.0",
"scipy", "numpy>=1.19.3",
"typeguard", "kaldi-native-fbank",
- "PyYAML>=5.1.2"],
- packages=['torch_paraformer'],
+ "PyYAML>=5.1.2", "torch-quant >= 0.4.0"],
+ packages=find_packages(include=["torch_paraformer*"]),
keywords=[
- 'funasr,paraformer'
+ 'funasr,paraformer, funasr_torch'
],
classifiers=[
'Programming Language :: Python :: 3.6',
--
Gitblit v1.9.1