From 7ffdd6c7d36c04f7a2bac579f8eba350450ab0cc Mon Sep 17 00:00:00 2001
From: shixian.shi <shixian.shi@alibaba-inc.com>
Date: 星期四, 12 十月 2023 11:41:04 +0800
Subject: [PATCH] update setup.py

---
 setup.py |    6 +++++-
 1 files changed, 5 insertions(+), 1 deletions(-)

diff --git a/setup.py b/setup.py
index 43c734c..2087c31 100644
--- a/setup.py
+++ b/setup.py
@@ -17,13 +17,14 @@
         "jamo",  # For kss
         "PyYAML>=5.1.2",
         "soundfile>=0.12.1",
-        "h5py>=2.10.0",
+        "h5py>=3.1.0",
         "kaldiio>=2.17.0",
         "torch_complex",
         "nltk>=3.4.5",
         # ASR
         "sentencepiece",
         "jieba",
+        "rotary_embedding_torch",
         # TTS
         "pypinyin>=0.44.0",
         "espnet_tts_frontend",
@@ -32,11 +33,14 @@
         "editdistance>=0.5.2",
         "tensorboard",
         "g2p",
+        "nara_wpe",
         # PAI
         "oss2",
         "edit-distance",
         "textgrid",
         "protobuf",
+        "tqdm",
+        "hdbscan",
     ],
     # train: The modules invoked when training only.
     "train": [

--
Gitblit v1.9.1