add curses-based interface for textual inversion

invoke-ai · Jan 4, 2023 · 1ff402f · 1ff402f
1 parent c9811b5
commit 1ff402f
Show file tree

Hide file tree

Showing 7 changed files with 251 additions and 156 deletions.
diff --git a/environments-and-requirements/requirements-base.txt b/environments-and-requirements/requirements-base.txt
@@ -37,6 +37,7 @@ test-tube>=0.7.5
 torch-fidelity
 torchmetrics
 transformers~=4.25
+windows-curses; sys_platform == 'win32'
 https://github.com/Birch-san/k-diffusion/archive/refs/heads/mps.zip#egg=k-diffusion
 https://github.com/invoke-ai/PyPatchMatch/archive/refs/tags/0.1.5.zip#egg=pypatchmatch
 https://github.com/openai/CLIP/archive/eaa22acb90a5876642d0507623e859909230a52d.zip#egg=clip

diff --git a/installer/templates/invoke.bat.in b/installer/templates/invoke.bat.in
@@ -9,15 +9,20 @@ set INVOKEAI_ROOT=.
 echo Do you want to generate images using the
 echo 1. command-line
 echo 2. browser-based UI
-echo 3. open the developer console
-set /P restore="Please enter 1, 2 or 3: "
+echo 3. textual inversion training
+echo 4. open the developer console
+set /P restore="Please enter 1, 2, 3 or 4: [1] "
+if not defined restore set restore=1
 IF /I "%restore%" == "1" (
     echo Starting the InvokeAI command-line..
     python .venv\Scripts\invoke.py %*
 ) ELSE IF /I "%restore%" == "2" (
     echo Starting the InvokeAI browser-based UI..
     python .venv\Scripts\invoke.py --web %*
 ) ELSE IF /I "%restore%" == "3" (
+    echo Starting textual inversion training..
+    python .venv\Scripts\textual_inversion_fe.py --web %*
+) ELSE IF /I "%restore%" == "4" (
     echo Developer Console
     echo Python command is:
     where python

diff --git a/installer/templates/invoke.sh.in b/installer/templates/invoke.sh.in
@@ -19,12 +19,15 @@ if [ "$0" != "bash" ]; then
     echo "Do you want to generate images using the"
     echo "1. command-line"
     echo "2. browser-based UI"
-    echo "3. open the developer console"
-    read -p "Please enter 1, 2, or 3: " yn
-    case $yn in
+    echo "3. textual inversion training"
+    echo "4. open the developer console"
+    read -p "Please enter 1, 2, 3 or 4: [1] " yn
+    choice=${yn:='1'}
+    case $choice in
         1 ) printf "\nStarting the InvokeAI command-line..\n"; .venv/bin/python .venv/bin/invoke.py  $*;;
         2 ) printf "\nStarting the InvokeAI browser-based UI..\n"; .venv/bin/python .venv/bin/invoke.py --web $*;;
-        3 ) printf "\nDeveloper Console:\n"; file_name=$(basename "${BASH_SOURCE[0]}"); bash --init-file "$file_name";;
+        3 ) printf "\nStarting Textual Inversion:\n"; .venv/bin/python .venv/bin/textual_inversion_fe.py $*;;
+	4 ) printf "\nDeveloper Console:\n"; file_name=$(basename "${BASH_SOURCE[0]}"); bash --init-file "$file_name";;
         * ) echo "Invalid selection"; exit;;
     esac
 else # in developer console