From 548153260b27b28bfdc880472e382e3418a05be3 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期四, 02 三月 2023 20:20:44 +0800
Subject: [PATCH] torchscripts
---
funasr/runtime/python/libtorch/setup.py | 4 ++--
1 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/funasr/runtime/python/libtorch/setup.py b/funasr/runtime/python/libtorch/setup.py
index 99d8b52..0f9e40d 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
@@ -29,7 +29,7 @@
"scipy", "numpy>=1.19.3",
"typeguard", "kaldi-native-fbank",
"PyYAML>=5.1.2"],
- packages=['torch_paraformer'],
+ packages=find_packages(include=["torch_paraformer*"]),
keywords=[
'funasr,paraformer'
],
--
Gitblit v1.9.1