From 339b06ca417bc09a10f722456f386886831e0801 Mon Sep 17 00:00:00 2001
From: hjh0119 <hujinghan.hjh@alibaba-inc.com>
Date: Mon, 7 Jul 2025 17:07:44 +0800
Subject: [PATCH] revert flash-attn dep

---
 docker/build_image.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docker/build_image.py b/docker/build_image.py
index 3f556ef3..bfef7be5 100644
--- a/docker/build_image.py
+++ b/docker/build_image.py
@@ -281,7 +281,7 @@ class LLMImageBuilder(Builder):
         if not args.autogptq_version:
             args.autogptq_version = '0.7.1'
         if not args.flashattn_version:
-            args.flashattn_version = '2.8.0.post2'
+            args.flashattn_version = '2.7.1.post4'
         return args
 
     def generate_dockerfile(self) -> str: