Merge pull request #258 from instructlab/mergify/bp/release-v0.5/pr-255

fix: Add explicit flash_attn requirement for ROCm (backport #255)
instructlab · Oct 8, 2024 · 097fdd0 · 097fdd0
2 parents ed833b9 + 4c05199
commit 097fdd0
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