Add a one-click setup and start script for Windows users (Akegarasu#22)

YunCheng66 · Mar 20, 2023 · 43d813c · 43d813c
1 parent 708492c
commit 43d813c
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -6,5 +6,6 @@ git/
 model/
 outputs/
 __pycache__/
+.venv/
 
 test.py
diff --git a/webui-start.cmd b/webui-start.cmd
@@ -0,0 +1,21 @@
+@echo off
+cd /D "%~dp0"
+if exist .venv goto :start
+
+echo "Setup VENV"
+python -m venv .venv
+call .venv\Scripts\activate.bat
+
+echo "Install dependencies"
+pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 --extra-index-url https://download.pytorch.org/whl/cu117
+pip install --upgrade -r requirements.txt
+goto :run
+
+:start
+echo "Start VENV"
+call .venv\Scripts\activate.bat
+goto :run
+
+:run
+echo "Start WebUI"
+python webui.py
-Original file line number
+Diff line change
@@ Expand Up / @@ -6,5 +6,6 @@ git/ @@
     model/
     outputs/
     __pycache__/
+    .venv/
     test.py