resolve python 3.8 package

Former-commit-id: 5eee4ec7016846356715a4fa1ad58e3cbb1cac6e
2024-05-09 16:52:27 +08:00
parent 4ce4172c87
commit 223bbd9930
5 changed files with 20 additions and 6 deletions
--- a/setup.py
+++ b/setup.py
@@ -20,6 +20,7 @@ def get_requires():


 extra_require = {
+    "torch": ["torch>=1.13.1"],
    "metrics": ["nltk", "jieba", "rouge-chinese"],
    "deepspeed": ["deepspeed>=0.10.0,<=0.14.0"],
    "bitsandbytes": ["bitsandbytes>=0.39.0"],