Merge pull request #546 from instructlab/mergify/bp/release-v0.10/pr-455

mergify[bot] · web-flow · commit 40e1e8c7c331 · 2025-05-08T19:44:06.000Z
moves deepspeed requirements into their own file; add deepspeed extras (backport #455)
diff --git a/pyproject.toml b/pyproject.toml
@@ -47,6 +47,7 @@ dependencies = { file = ["requirements.txt"] }
 optional-dependencies.cuda = { file = ["requirements-cuda.txt"] }
 optional-dependencies.rocm = { file = ["requirements-rocm.txt"] }
 optional-dependencies.hpu = { file = ["requirements-hpu.txt"] }
+optional-dependencies.deepspeed = { file = ["requirements-deepspeed.txt"] }
 
 [tool.setuptools.packages.find]
 where = ["src"]
diff --git a/requirements-cuda.txt b/requirements-cuda.txt
@@ -1,9 +1,6 @@
 flash-attn>=2.4.0
 bitsandbytes>=0.43.1
 
-# available as an option for NVIDIA, FSDP still default
-deepspeed>=0.14.3
-
 # required for FSDP updates
 accelerate>=0.34.2,<1.1.0
 
diff --git a/requirements-deepspeed.txt b/requirements-deepspeed.txt
@@ -0,0 +1,3 @@
+# SPDX-License-Identifier: Apache-2.0
+
+deepspeed>=0.14.3

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+# SPDX-License-Identifier: Apache-2.0`
	`2`	`+`
	`3`	`+deepspeed>=0.14.3`