Ver Fonte

Optimize deps & add optional asr

Lengyue há 2 anos atrás
pai
commit
675aea0519
1 ficheiros alterados com 13 adições e 10 exclusões
  1. 13 10
      pyproject.toml

+ 13 - 10
pyproject.toml

@@ -15,8 +15,6 @@ classifiers = [
 dependencies = [
 dependencies = [
     "transformers>=4.35.2",
     "transformers>=4.35.2",
     "datasets>=2.14.5",
     "datasets>=2.14.5",
-    "bitsandbytes>=0.41.1",
-    "peft>=0.5.0",
     "lightning>=2.1.0",
     "lightning>=2.1.0",
     "hydra-core>=1.3.2",
     "hydra-core>=1.3.2",
     "tensorboard>=2.14.1",
     "tensorboard>=2.14.1",
@@ -26,17 +24,22 @@ dependencies = [
     "vector-quantize-pytorch>=1.10.0",
     "vector-quantize-pytorch>=1.10.0",
     "rich>=13.5.3",
     "rich>=13.5.3",
     "gradio>=4.0.0",
     "gradio>=4.0.0",
-    "cn2an",
-    "pypinyin",
-    "jieba",
-    "g2p_en",
-    "pyopenjtalk",
-    "wandb",
-    "tensorboard",
+    "cn2an>=0.5.22",
+    "pypinyin>=0.49.0",
+    "jieba>=0.42.1",
+    "g2p-en>=2.1.0",
+    "pyopenjtalk>=0.3.2",
+    "wandb>=0.15.11",
     "grpcio>=1.58.0",
     "grpcio>=1.58.0",
     "kui>=1.6.0",
     "kui>=1.6.0",
     "zibai-server>=0.9.0",
     "zibai-server>=0.9.0",
-    "loguru",
+    "loguru>=0.6.0",
+]
+
+[project.optional-dependencies]
+asr = [
+    "openai-whisper",
+    "modelscope"
 ]
 ]
 
 
 [build-system]
 [build-system]