Update scripts/mainrunpodA1111.py

scripts/mainrunpodA1111.py

@@ -318,7 +318,48 @@ def loradwn(LoRA_LINK):
         else:
           print('Wrong link, check that the link is valid')
 
+def InstallDeforum():   
+    os.chdir('/workspace/sd/stable-diffusion-webui/extensions')
+    if not os.path.exists("deforum-for-automatic1111-webui"):
+      call('git clone https://github.com/deforum-art/sd-webui-deforum.git', shell=True)
+      os.chdir('/workspace')
+    else:
+      os.chdir('deforum-for-automatic1111-webui')
+      call('git reset --hard', shell=True, stdout=open('/dev/null', 'w'), stderr=open('/dev/null', 'w'))
+      call('git pull', shell=True, stdout=open('/dev/null', 'w'), stderr=open('/dev/null', 'w'))
+      os.chdir('/workspace')
+
+def InstallWildCards():   
+    # Путь к целевой директории
+    target_dir = '/workspace/sd/stable-diffusion-webui/extensions/stable-diffusion-webui-wildcards/wildcards/'
+
+    # Перемещаемся в директорию расширений
+    os.chdir('/workspace/sd/stable-diffusion-webui/extensions')
+
+    if not os.path.exists("stable-diffusion-webui-wildcards"):
+        # Клонирование репозитория, если он ещё не существует
+        call('git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui-wildcards.git', shell=True)
+        
+        # Перемещаемся в клонированный репозиторий
+        os.chdir('stable-diffusion-webui-wildcards')
+
+        # Скачиваем файл по ID
+        gdown.download(id='1sY9Yv29cCYZuxBvszkmLVgw--aRdwT1P', output='wildcards.zip', quiet=False)
 
+        # Создаем целевую директорию, если она ещё не существует
+        os.makedirs(target_dir, exist_ok=True)
+
+        # Распаковываем архив
+        with zipfile.ZipFile('wildcards.zip', 'r') as zip_ref:
+            zip_ref.extractall(target_dir)
+    else:
+        # Если репозиторий уже существует, обновляем его
+        os.chdir('stable-diffusion-webui-wildcards')
+        call('git reset --hard', shell=True, stdout=open(os.devnull, 'w'), stderr=open(os.devnull, 'w'))
+        call('git pull', shell=True, stdout=open(os.devnull, 'w'), stderr=open(os.devnull, 'w'))
+
+    # Возвращаемся в исходную рабочую директорию
+    os.chdir('/workspace')
 
 def CNet(ControlNet_Model, ControlNet_XL_Model):