~

files_cells/notebooks/en/downloading_en.ipynb

@@ -50,6 +50,16 @@
         "#  ----------------------------------------------\n",
         "\n",
         "\n",
+        "# == ONLY SAGEMAKER ===\n",
+        "if env == \"SageMaker Studio Lab\":\n",
+        "    print(\"Обновление зависимостей, может занять некоторое время...\")\n",
+        "    !pip install -q --upgrade torchsde\n",
+        "    !pip install -q --upgrade pip\n",
+        "    !pip install -q --upgrade psutil\n",
+        "\n",
+        "    clear_output()\n",
+        "\n",
+        "\n",
         "# ================ LIBRARIES ================\n",
         "flag_file = f\"{root_path}/libraries_installed.txt\"\n",
         "\n",
@@ -65,6 +75,11 @@
         "        !curl -s -Lo /usr/bin/cl https://github.com/cloudflare/cloudflared/releases/latest/download/cloudflared-linux-amd64 && chmod +x /usr/bin/cl\n",
         "        !pip install insightface\n",
         "\n",
+        "        if env == \"SageMaker Studio Lab\":\n",
+        "            %pip install -q opencv-python-headless huggingface-hub\n",
+        "            %conda update -q -n base conda\n",
+        "            %conda install -q -y glib\n",
+        "\n",
         "        if env == \"Google Colab\":\n",
         "            !pip install -q xformers==0.0.22.post7\n",
         "        else:\n",
@@ -158,9 +173,14 @@
         "\n",
         "        ## Update extensions\n",
         "        if latest_exstensions:\n",
-        "            for d in os.listdir(f'{webui_path}/extensions/'):\n",
-        "                if os.path.isdir(f'{webui_path}/extensions/{d}'):\n",
-        "                    os.system(f'cd {webui_path}/extensions/{d} && git reset --hard && git pull')\n",
+        "            extensions_path = f\"{webui_path}/extensions/\"\n",
+        "\n",
+        "            for dir_name in os.listdir(extensions_path):\n",
+        "                repo_path = os.path.join(extensions_path, dir_name)\n",
+        "                if os.path.isdir(repo_path):\n",
+        "                    %cd {repo_path}\n",
+        "                    !git reset --hard\n",
+        "                    !git pull\n",
         "\n",
         "        !cd {webui_path}/repositories/stable-diffusion-stability-ai && git restore .\n",
         "    del cap\n",

files_cells/notebooks/ru/downloading_ru.ipynb

@@ -65,6 +65,11 @@
         "        !curl -s -Lo /usr/bin/cl https://github.com/cloudflare/cloudflared/releases/latest/download/cloudflared-linux-amd64 && chmod +x /usr/bin/cl\n",
         "        !pip install insightface\n",
         "\n",
+        "        if env == \"SageMaker Studio Lab\":\n",
+        "            %pip install -q opencv-python-headless huggingface-hub\n",
+        "            %conda update -q -n base conda\n",
+        "            %conda install -q -y glib\n",
+        "\n",
         "        if env == \"Google Colab\":\n",
         "            !pip install -q xformers==0.0.22.post7\n",
         "        else:\n",
@@ -158,9 +163,14 @@
         "\n",
         "        ## Update extensions\n",
         "        if latest_exstensions:\n",
-        "            for d in os.listdir(f'{webui_path}/extensions/'):\n",
-        "                if os.path.isdir(f'{webui_path}/extensions/{d}'):\n",
-        "                    os.system(f'cd {webui_path}/extensions/{d} && git reset --hard && git pull')\n",
+        "            extensions_path = f\"{webui_path}/extensions/\"\n",
+        "\n",
+        "            for dir_name in os.listdir(extensions_path):\n",
+        "                repo_path = os.path.join(extensions_path, dir_name)\n",
+        "                if os.path.isdir(repo_path):\n",
+        "                    %cd {repo_path}\n",
+        "                    !git reset --hard\n",
+        "                    !git pull\n",
         "\n",
         "        !cd {webui_path}/repositories/stable-diffusion-stability-ai && git restore .\n",
         "    del cap\n",

files_cells/python/en/downloading_en.py

@@ -37,6 +37,16 @@ webui_path = f"{root_path}/sdw"
 #  ----------------------------------------------
 
 
+# == ONLY SAGEMAKER ===
+if env == "SageMaker Studio Lab":
+    print("Обновление зависимостей, может занять некоторое время...")
+    get_ipython().system('pip install -q --upgrade torchsde')
+    get_ipython().system('pip install -q --upgrade pip')
+    get_ipython().system('pip install -q --upgrade psutil')
+
+    clear_output()
+
+
 # ================ LIBRARIES ================
 flag_file = f"{root_path}/libraries_installed.txt"
 
@@ -52,6 +62,11 @@ if not os.path.exists(flag_file):
         get_ipython().system('curl -s -Lo /usr/bin/cl https://github.com/cloudflare/cloudflared/releases/latest/download/cloudflared-linux-amd64 && chmod +x /usr/bin/cl')
         get_ipython().system('pip install insightface')
 
+        if env == "SageMaker Studio Lab":
+            get_ipython().run_line_magic('pip', 'install -q opencv-python-headless huggingface-hub')
+            get_ipython().run_line_magic('conda', 'update -q -n base conda')
+            get_ipython().run_line_magic('conda', 'install -q -y glib')
+
         if env == "Google Colab":
             get_ipython().system('pip install -q xformers==0.0.22.post7')
         else:
@@ -145,9 +160,14 @@ if latest_webui or latest_exstensions:
 
         ## Update extensions
         if latest_exstensions:
-            for d in os.listdir(f'{webui_path}/extensions/'):
-                if os.path.isdir(f'{webui_path}/extensions/{d}'):
-                    os.system(f'cd {webui_path}/extensions/{d} && git reset --hard && git pull')
+            extensions_path = f"{webui_path}/extensions/"
+
+            for dir_name in os.listdir(extensions_path):
+                repo_path = os.path.join(extensions_path, dir_name)
+                if os.path.isdir(repo_path):
+                    get_ipython().run_line_magic('cd', '{repo_path}')
+                    get_ipython().system('git reset --hard')
+                    get_ipython().system('git pull')
 
         get_ipython().system('cd {webui_path}/repositories/stable-diffusion-stability-ai && git restore .')
     del cap

files_cells/python/ru/downloading_ru.py

@@ -52,6 +52,11 @@ if not os.path.exists(flag_file):
         get_ipython().system('curl -s -Lo /usr/bin/cl https://github.com/cloudflare/cloudflared/releases/latest/download/cloudflared-linux-amd64 && chmod +x /usr/bin/cl')
         get_ipython().system('pip install insightface')
 
+        if env == "SageMaker Studio Lab":
+            get_ipython().run_line_magic('pip', 'install -q opencv-python-headless huggingface-hub')
+            get_ipython().run_line_magic('conda', 'update -q -n base conda')
+            get_ipython().run_line_magic('conda', 'install -q -y glib')
+
         if env == "Google Colab":
             get_ipython().system('pip install -q xformers==0.0.22.post7')
         else:
@@ -145,9 +150,14 @@ if latest_webui or latest_exstensions:
 
         ## Update extensions
         if latest_exstensions:
-            for d in os.listdir(f'{webui_path}/extensions/'):
-                if os.path.isdir(f'{webui_path}/extensions/{d}'):
-                    os.system(f'cd {webui_path}/extensions/{d} && git reset --hard && git pull')
+            extensions_path = f"{webui_path}/extensions/"
+
+            for dir_name in os.listdir(extensions_path):
+                repo_path = os.path.join(extensions_path, dir_name)
+                if os.path.isdir(repo_path):
+                    get_ipython().run_line_magic('cd', '{repo_path}')
+                    get_ipython().system('git reset --hard')
+                    get_ipython().system('git pull')
 
         get_ipython().system('cd {webui_path}/repositories/stable-diffusion-stability-ai && git restore .')
     del cap