InstallReActor

scripts/mainrunpodA1111.py

@@ -442,7 +442,41 @@ def InstallDeforum():
         # call('git pull', shell=True, stdout=open(os.devnull, 'w'), stderr=open(os.devnull, 'w'))
         # Возврат в корневую директорию
         os.chdir("/workspace")
-
+        
+def InstallReActor():
+    # Переход в директорию расширений
+    os.chdir("/workspace/sd/stable-diffusion-webui/extensions")
+    # Проверка наличия директории с расширением
+    if not os.path.exists("sd-webui-reactor"):
+        # Установка библиотек
+        from subprocess import call, DEVNULL
+        call(
+            'pip install "onnxruntime-gpu>=1.16.1"', 
+             shell=True
+        )
+        # Клонирование репозитория, если он не существует
+        call(
+            "git clone https://github.com/Gourieff/sd-webui-reactor.git",
+            shell=True,
+        )
+        # Создание папки под модели
+        call(
+            "mkdir /workspace/sd/stable-diffusion-webui/models/insightface",
+            shell=True
+        )
+        os.chdir("/workspace/sd/stable-diffusion-webui/models/insightface")
+        # Скачивание модели для reActor
+        call(
+            "wget https://github.com/facefusion/facefusion-assets/releases/download/models/inswapper_128.onnx", 
+            shell=True, 
+            stdout=DEVNULL, 
+            stderr=DEVNULL
+        )
+        # Вовзрат в корневую директорию
+        os.chdir("/workspace")
+    else:
+        Возврат в корневую директорию
+        os.chdir("/workspace")
 
 def InstallAnimateDiff():
     # Переход в директорию расширений
@@ -450,6 +484,7 @@ def InstallAnimateDiff():
 
     # Проверка наличия директории с расширением
     if not os.path.exists("sd-webui-animatediff"):
+        from subprocess import call, DEVNULL
         # Клонирование репозитория, если он не существует
         call(
             "git clone https://github.com/continue-revolution/sd-webui-animatediff.git",
@@ -462,12 +497,16 @@ def InstallAnimateDiff():
         # Скачивание моделей для animatediff
         call(
             "wget https://huggingface.co/guoyww/animatediff/resolve/refs%2Fpr%2F3/mm_sd_v15_v2.ckpt",
-            shell=True,
-        )
-        call(
-            "wget https://huggingface.co/manshoety/AD_Stabilized_Motion/resolve/main/mm-Stabilized_high.pth",
-            shell=True,
+            shell=True, 
+            stdout=DEVNULL, 
+            stderr=DEVNULL
         )
+      #  call(
+      #      "wget https://huggingface.co/manshoety/AD_Stabilized_Motion/resolve/main/mm-Stabilized_high.pth",
+      #      shell=True, 
+      #      stdout=DEVNULL, 
+      #      stderr=DEVNULL
+      #  )
         # Вовзрат в корневую директорию
         os.chdir("/workspace")
     else: