fix: Add explicit flash_attn requirement for ROCm

Signed-off-by: Fabien Dupont <[email protected]> (cherry picked from commit d74105a)
instructlab · Oct 8, 2024 · 4c05199 · 4c05199
1 parent ed833b9
commit 4c05199
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/requirements-rocm.txt b/requirements-rocm.txt
@@ -0,0 +1 @@
+flash-attn>=2.6.2,<2.7.0