CLIP interrogator

1 files changed, 10 insertions, 3 deletions

diff --git a/webui.bat b/webui.bat
index 1b973448..54734d07 100644
--- a/webui.bat
+++ b/webui.bat
@@ -85,7 +85,7 @@ if %ERRORLEVEL% == 0 goto :install_reqs
 goto :show_stdout_stderr

 

 :install_reqs

-%PYTHON% -c "import omegaconf; import fonts" >tmp/stdout.txt 2>tmp/stderr.txt

+%PYTHON% -c "import omegaconf; import fonts; import timm" >tmp/stdout.txt 2>tmp/stderr.txt

 if %ERRORLEVEL% == 0 goto :make_dirs

 echo Installing requirements...

 %PYTHON% -m pip install -r %REQS_FILE% --prefer-binary >tmp/stdout.txt 2>tmp/stderr.txt

@@ -117,12 +117,19 @@ goto :show_stdout_stderr
 

 :install_codeformer_reqs

 %PYTHON% -c "import lpips" >tmp/stdout.txt 2>tmp/stderr.txt

-if %ERRORLEVEL% == 0 goto :check_model

+if %ERRORLEVEL% == 0 goto :clone_blip

 echo Installing requirements for CodeFormer...

 %PYTHON% -m pip install -r repositories\CodeFormer\requirements.txt --prefer-binary >tmp/stdout.txt 2>tmp/stderr.txt

-if %ERRORLEVEL% == 0 goto :check_model

+if %ERRORLEVEL% == 0 goto :clone_blip

 goto :show_stdout_stderr

 

+:clone_blip

+if exist repositories\BLIP goto :check_model

+echo Cloning BLIP repository...

+%GIT% clone https://github.com/salesforce/BLIP.git repositories\BLIP >tmp/stdout.txt 2>tmp/stderr.txt

+if %ERRORLEVEL% NEQ 0 goto :show_stdout_stderr

+%GIT% -C repositories/BLIP checkout 48211a1594f1321b00f14c9f7a5b4813144b2fb9 >tmp/stdout.txt 2>tmp/stderr.txt

+if %ERRORLEVEL% NEQ 0 goto :show_stdout_stderr

 

 :check_model

 dir model.ckpt >tmp/stdout.txt 2>tmp/stderr.txt