fix: check test_mode

wy-luke · Sep 5, 2023 · 37416b1 · 37416b1
1 parent 5f98696
commit 37416b1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/sd-installer.sh b/sd-installer.sh
@@ -266,12 +266,12 @@ fi
 echo "############ 开始安装 Stable Diffusion ################"
 echo "############ Start to install Stable Diffusion ######"
 
-if $test_mode; then
+if "$test_mode" != 0; then
     if ./webui.sh | grep -q 'Launching Web UI with arguments: '; then
-        clean_up
-        echo "############ Install Stable Diffusion successfully ###"
         exit
     fi
 else
     ./webui.sh
+    clean_up
+    echo "############ Install Stable Diffusion successfully ###"
 fi