From 37b6a42aa72df0399a43d3f6617d43cbf86cb01d Mon Sep 17 00:00:00 2001 From: lonestriker Date: Tue, 31 Oct 2023 06:55:27 +0400 Subject: [PATCH] Add flash-attn to requirements.txt --- requirements.txt | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index 5b17b57..b8bb2e0 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,4 +2,5 @@ torch>=2.1.0 pynvml exllamav2>=0.0.7 Flask>=2.3.2 -waitress>=2.1.2 \ No newline at end of file +waitress>=2.1.2 +flash-attn>=2.3.3