From a4a3b618e733ccf1f11de3d783879b0530d123c1 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Wed, 4 Mar 2026 10:00:08 -0500 Subject: [PATCH] force torch to match when installing fa and deepspeed using uv --- pyproject.toml | 2 ++ 1 file changed, 2 insertions(+) diff --git a/pyproject.toml b/pyproject.toml index e60f6f3ff..bd184c49e 100644 --- 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 }]