update

2025-08-30 00:42:38 -04:00
parent 1334281d50
commit d1fd505813
5 changed files with 14 additions and 21 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -101,6 +101,8 @@ dependencies = [
    "torchao==0.12.0 ; sys_platform != 'darwin'",
    "bitsandbytes==0.47.0 ; sys_platform != 'darwin'",
    "flash-attn==2.8.3 ; sys_platform == 'linux'",
+    "deepspeed>=0.17.5 ; sys_platform != 'darwin'",
+    "deepspeed-kernels ; sys_platform != 'darwin'",
 ]

 [project.optional-dependencies]
@@ -110,12 +112,6 @@ ring-flash-attn = [
    "yunchang==0.6.0",
 ]

-# Deep learning frameworks
-deepspeed = [
-    "deepspeed>=0.17.5",
-    "deepspeed-kernels",
-]
-
 mamba-ssm = [
    "mamba-ssm>=2.2.0",
    "causal_conv1d>=1.4.0",