From ef1063a3e909872096af444952a4c6afb4c256a8 Mon Sep 17 00:00:00 2001
From: Lai HaoWen <createskyblue@outlook.com>
Date: Sat, 31 May 2025 03:47:14 +0000
Subject: [PATCH] Add proxy support for flash-attention git clone

- Modified git clone command to use GHPROXY_URL when USE_GHPROXY=1
- Added debug logging to show which URL is being used
- Maintained existing error handling and fallback behavior
- Improved code structure for better maintainability
---
 run.sh | 19 +++++++++----------
 1 file changed, 9 insertions(+), 10 deletions(-)

diff --git a/run.sh b/run.sh
index 718eea5..dd70cde 100644
--- a/run.sh
+++ b/run.sh
@@ -2566,8 +2566,16 @@ install_flash_attn() {
         log "ERROR" "无法创建临时目录"
         return 1
     }
+    
+    local flash_attn_url="https://github.com/Dao-AILab/flash-attention.git"
+    if [ $USE_GHPROXY -eq 1 ]; then
+        flash_attn_url="${GHPROXY_URL}/${flash_attn_url}"
+        echo -e "${YELLOW}使用代理克隆flash-attention: ${flash_attn_url}${NC}"
+    else
+        echo -e "${YELLOW}直接克隆flash-attention: ${flash_attn_url}${NC}"
+    fi
 
-    if git clone https://github.com/Dao-AILab/flash-attention.git; then
+    if git clone "$flash_attn_url"; then
         cd flash-attention || {
             log "ERROR" "无法进入flash-attention目录"
             return 1
@@ -2991,12 +2999,3 @@ main() {
 
 # 运行主函数
 main "$@"
-
-
-
-
-
-
-
-
-