force torch to match when installing fa and deepspeed using uv

2026-03-04 10:00:08 -05:00
1 changed files with 2 additions and 0 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -63,3 +63,5 @@ docstring-code-format = false

 [tool.uv.extra-build-dependencies]
 axolotl = ["huggingface_hub"]
+flash-attn = [{ requirement = "torch", match-runtime = true }]
+deepspeed = [{ requirement = "torch", match-runtime = true }]