resolve python 3.8 package

2025-12-15 03:10:35 +08:00 · 2024-05-09 16:52:27 +08:00
parent d9cdddd19c
commit 75aec4cf8e
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"],