fixing eval for SP (#2468)

2025-04-01 11:59:08 -04:00
parent df119e3724
commit 7d0eb66b54
3 changed files with 57 additions and 27 deletions
--- a/setup.py
+++ b/setup.py
@@ -106,7 +106,11 @@ def get_package_version():

 extras_require = {
    "flash-attn": ["flash-attn==2.7.4.post1"],
-    "ring-flash-attn": ["ring-flash-attn>=0.1.4", "yunchang==0.6.0"],
+    "ring-flash-attn": [
+        "flash-attn==2.7.4.post1",
+        "ring-flash-attn>=0.1.4",
+        "yunchang==0.6.0",
+    ],
    "deepspeed": [
        "deepspeed==0.16.4",
        "deepspeed-kernels",