From f194457229e4537912467bc60ac3a873f473a63c Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Sun, 11 Sep 2022 18:48:36 +0300
Subject: CLIP interrogator

---
 webui.bat | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

(limited to 'webui.bat')

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
-- 
cgit v1.2.1