Update scripts/mainrunpodA1111.py

scripts/mainrunpodA1111.py

@@ -22,12 +22,19 @@ import zipfile
 
 
 def Deps(force_reinstall):
-
-    if not force_reinstall and os.path.exists('/usr/local/lib/python3.10/dist-packages/safetensors'):
+    # Определяем путь к файлу-маркеру
+    marker_file = '/workspace/setup_complete'
+    if not force_reinstall and not os.path.exists(marker_file):
         ntbks()
         print('Modules and notebooks updated, dependencies already installed')
         os.environ['TORCH_HOME'] = '/workspace/cache/torch'
         os.environ['PYTHONWARNINGS'] = 'ignore'
+        call("pip install controlnet_aux --root-user-action=ignore --disable-pip-version-check", shell=True, stdout=open('/dev/null', 'w'))
+        call("pip install diskcache --root-user-action=ignore --disable-pip-version-check", shell=True, stdout=open('/dev/null', 'w'))
+        call("pip install pillow-avif-plugin --root-user-action=ignore --disable-pip-version-check", shell=True, stdout=open('/dev/null', 'w'))
+        call('pip install --root-user-action=ignore --disable-pip-version-check -qq gradio==3.41.2', shell=True, stdout=open('/dev/null', 'w'))
+        clear_output()
+        done()
     else:
         call('pip install --root-user-action=ignore --disable-pip-version-check --no-deps -qq gdown PyWavelets numpy==1.23.5 accelerate==0.12.0 --force-reinstall', shell=True, stdout=open('/dev/null', 'w'))
         ntbks()
@@ -54,7 +61,6 @@ def Deps(force_reinstall):
         os.environ['PYTHONWARNINGS'] = 'ignore'
         call("sed -i 's@text = _formatwarnmsg(msg)@text =\"\"@g' /usr/lib/python3.10/warnings.py", shell=True)
         clear_output()
-
         done()