From 4c05199fe9bb36295ff7a02ce585713bc4c5c541 Mon Sep 17 00:00:00 2001
From: Fabien Dupont <fdupont@redhat.com>
Date: Mon, 7 Oct 2024 16:07:59 -0400
Subject: [PATCH] fix: Add explicit flash_attn requirement for ROCm

Signed-off-by: Fabien Dupont <fdupont@redhat.com>
(cherry picked from commit d74105a85991edea5b47c8a8f8fc6dd8df7a7edc)
---
 requirements-rocm.txt | 1 +
 1 file changed, 1 insertion(+)
 create mode 100644 requirements-rocm.txt

diff --git a/requirements-rocm.txt b/requirements-rocm.txt
new file mode 100644
index 00000000..21612c36
--- /dev/null
+++ b/requirements-rocm.txt
@@ -0,0 +1 @@
+flash-attn>=2.6.2,<2.7.0