From 1e2cd05bdaa167bccbad8453d694ab3cfb768c8a Mon Sep 17 00:00:00 2001
From: ver217 <lhx0217@gmail.com>
Date: Wed, 16 Aug 2023 10:44:16 +0800
Subject: [PATCH] [misc] fix flash attn requirement

---
 requirements/requirements-test.txt | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/requirements/requirements-test.txt b/requirements/requirements-test.txt
index 1a2d8cbff625..57a354d0d013 100644
--- a/requirements/requirements-test.txt
+++ b/requirements/requirements-test.txt
@@ -16,4 +16,4 @@ triton==2.0.0.dev20221202
 requests==2.27.1 # downgrade to avoid huggingface error https://github.com/huggingface/transformers/issues/17611
 SentencePiece
 ninja
-flash-attn
+flash-attn==2.0.5