From adb12b79f6f46f708e079a8c67563dfbdd42b465 Mon Sep 17 00:00:00 2001
From: wy-luke <li1078797083@qq.com>
Date: Tue, 5 Sep 2023 14:33:18 +0800
Subject: [PATCH] feat: add ali pip mirror

---
 sd-installer.sh | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/sd-installer.sh b/sd-installer.sh
index ea5226e..167e037 100644
--- a/sd-installer.sh
+++ b/sd-installer.sh
@@ -249,11 +249,13 @@ source venv/bin/activate
 # Delete pip cache to avoid some errors
 pip cache purge
 
-# Tsinghua mirror has no nightly package, which is needed by basicsr
-# if ! $net_connected; then
-#     pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
-#     pip install torch
-# fi
+if ! $net_connected; then
+    # Tsinghua mirror has no tb-nightly package, which is needed by basicsr
+    pip config set global.index-url https://mirrors.aliyun.com/pypi/simple
+    pip config set global.extra-index-url https://pypi.tuna.tsinghua.edu.cn/simple
+
+    pip install torch
+fi
 
 pip install --upgrade pip setuptools
 pip install basicsr==1.4.2