Merge pull request #80 from houseofsecrets/develop

Updated img2img start scripts with the --source argument
houseofsecrets · Nov 11, 2023 · 097b8da · 097b8da
2 parents 23e4926 + b05b2c1
commit 097b8da
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/Start_img2img.bat b/Start_img2img.bat
@@ -11,13 +11,14 @@ REM Activate the virtual environment
 call venv/Scripts/activate.bat
 
 set "var=%~1"
+if "!var!" == "" set "var=#"
 
 REM Install required packages
 pip install -r requirements.txt
 
 REM Run the script
 
-python SdPaint.py --img2img "!var!"
+python SdPaint.py --img2img --source "!var!"
 
 REM Deactivate the virtual environment
 deactivate
diff --git a/start-img2img.sh b/start-img2img.sh
@@ -2,10 +2,19 @@
 
 [ -d venv ] || python -m venv venv
 
-source ./venv/bin/activate
+if [[ -d  ./venv/Scripts/ ]]; then
+    source ./venv/Scripts/activate
+else
+  source ./venv/bin/activate
+fi
+
+SOURCE="${@}"
+if [[ -z "${SOURCE}" ]]; then
+    SOURCE="#"
+fi
 
 pip install -r requirements.txt
 
-python SdPaint.py --img2img "$@"
+python SdPaint.py --img2img --source "${SOURCE}"
 
 deactivate