Update scripts/mainrunpodA1111.py

scripts/mainrunpodA1111.py

@@ -25,12 +25,52 @@ def Deps(force_reinstall):
     import os
     from subprocess import call
     import venv
+    
     if not force_reinstall and os.path.exists('/usr/local/lib/python3.10/dist-packages/safetensors'):
+        # Активация виртуального окружения и обновление pip
+        activate_script = os.path.join(venv_dir, 'bin', 'activate')
+        call(f'/bin/bash -c "source {activate_script} && pip install --upgrade pip"', shell=True)
+        
         ntbks()
         print('Modules and notebooks updated, dependencies already installed')
         os.environ['TORCH_HOME'] = '/workspace/cache/torch'
         os.environ['PYTHONWARNINGS'] = 'ignore'
     else:
+
+        # Активация виртуального окружения и установка зависимостей из файла requirements.txt
+        requirements = """
+        fastapi==0.87.0
+        onnxruntime==1.13.1
+        scipy==1.9.3
+        uvicorn==0.20.0
+        rembg==2.0.30
+        future
+        lmdb
+        google-auth<3,>=1.6.3
+        google-auth-oauthlib<0.5,>=0.4.1
+        grpcio>=1.24.3
+        markdown>=2.6.8
+        tensorboard-data-server<0.7.0,>=0.6.0
+        tensorboard-plugin-wit>=1.6.0
+        werkzeug>=1.0.1
+        requests==2.28.1
+        protobuf<4,>=3.20.2
+        gradio==3.41.2
+        gradio-client==0.5.0
+        huggingface-hub>=0.14.0
+        diskcache
+        pillow-avif-plugin
+        tomesd
+        pyre-extensions==0.0.29
+        """
+        
+        # Запись требований в файл requirements.txt
+        with open("/workspace/requirements.txt", "w") as req_file:
+            req_file.write(requirements)
+            
+        # Установка зависимостей из requirements.txt
+        call(f'/bin/bash -c "source {activate_script} && pip install -r /workspace/requirements.txt"', shell=True)
+        
         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'))
         if os.path.exists('deps'):
             call("rm -r deps", shell=True)
@@ -54,10 +94,10 @@ def Deps(force_reinstall):
         
         # Проверка, что файл существует
         if not os.path.exists("/workspace/deps/rnpddeps-t2.tar.zst"):
-            raise FileNotFoundError("The file /workspace/deps/rnpd-310.tar.zst does not exist")
+            raise FileNotFoundError("The file /workspace/deps/rnpddeps-t2.tar.zst does not exist")
         
         # Распаковка файла
-        call('tar -C / --zstd -xf /workspace/deps/rnpd-310.tar.zst', shell=True, stdout=open('/dev/null', 'w'))
+        call('tar -C / --zstd -xf /workspace/deps/rnpddeps-t2.tar.zst.zst', shell=True, stdout=open('/dev/null', 'w'))
         
         # Проверка наличия файла перед использованием sed
         hub_py_path = "/usr/local/lib/python3.10/dist-packages/transformers/utils/hub.py"