fixing extensions?

files_cells/notebooks/en/downloading_en.ipynb

@@ -80,6 +80,7 @@
         "            %conda update -q -n base conda\n",
         "            %conda install -q -y aria2\n",
         "            %conda install -q -y glib\n",
+        "            !pip install tensorflow\n",
         "\n",
         "            !wget -P /home/studio-lab-user https://huggingface.co/NagisaNao/fast_repo/resolve/main/sagemaker/FULL_DELETED_NOTEBOOK.ipynb\n",
         "\n",
@@ -160,6 +161,12 @@
         "    print(f\"⌚️ You have been conducting this session for - \\033[33m{time_since_start}\\033[0m\")\n",
         "\n",
         "\n",
+        "# === ONLY SAGEMAKER FIXIN EXTENSIONS ===\n",
+        "if env == \"SageMaker Studio Lab\":\n",
+        "    # --- SuperMerger ---\n",
+        "    !wget -O {webui_path}/extensions/supermerger/scripts/mergers/mergers.py https://huggingface.co/NagisaNao/fast_repo/resolve/main/sagemaker/fixing/extensions/supermerger/scripts/mergers/mergers.py\n",
+        "\n",
+        "\n",
         "## Changes extensions and WebUi\n",
         "if latest_webui or latest_exstensions:\n",
         "    action = \"Updating WebUI and Extensions\" if latest_webui and latest_exstensions else (\"WebUI Update\" if latest_webui else \"Update Extensions\")\n",
@@ -176,7 +183,10 @@
         "\n",
         "        ## Update extensions\n",
         "        if latest_exstensions:\n",
-        "            !{'for dir in ' + webui_path + '/extensions/*/; do cd \\\"$dir\\\" && git reset --hard && git pull; done'}\n",
+        "            if env != \"SageMaker Studio Lab\":\n",
+        "                !{'for dir in ' + webui_path + '/extensions/*/; do cd \\\"$dir\\\" && git reset --hard && git pull; done'}\n",
+        "            else:\n",
+        "                !{'for dir in /home/studio-lab-user/content/sdw/extensions/*/; do cd \\\"$dir\\\" && git fetch origin && git pull; done'}\n",
         "\n",
         "        !cd {webui_path}/repositories/stable-diffusion-stability-ai && git restore .\n",
         "    del cap\n",
@@ -463,7 +473,18 @@
         "!find {extensions_dir}/ -mindepth 1 -maxdepth 1 ! -name '*.txt' -printf '%f\\n'\n",
         "if any(not file.endswith(('.txt', '.yaml')) for file in os.listdir(control_dir)):\n",
         "    print(\"\\n\\033[33m➤ ControlNet\\033[0m\")\n",
-        "    !find {control_dir}/ -mindepth 1 ! -name '*.yaml' -printf '%f\\n' | sed 's/^[^_]*_[^_]*_[^_]*_\\(.*\\)_fp16\\.safetensors$/\\1/'"
+        "    !find {control_dir}/ -mindepth 1 ! -name '*.yaml' -printf '%f\\n' | sed 's/^[^_]*_[^_]*_[^_]*_\\(.*\\)_fp16\\.safetensors$/\\1/'\n",
+        "\n",
+        "\n",
+        "# === ONLY SAGEMAKER ===\n",
+        "if env == \"SageMaker Studio Lab\":\n",
+        "    with capture.capture_output() as cap:\n",
+        "        !rm -rf /home/studio-lab-user/.conda/envs/studiolab-safemode\n",
+        "        !rm -rf /home/studio-lab-user/.conda/envs/sagemaker-distribution\n",
+        "        !rm -rf /home/studio-lab-user/.conda/pkgs/cache\n",
+        "        !pip cache purge\n",
+        "        !rm -rf ~/.cache\n",
+        "    del cap"
       ],
       "metadata": {
         "id": "2lJmbqrs3Mu8"

files_cells/notebooks/en/launch_en.ipynb

@@ -87,6 +87,8 @@
         "# automatic fixing path V2\n",
         "!sed -i 's#\"tagger_hf_cache_dir\": \".*models/interrogators\"#\"tagger_hf_cache_dir\": \"{root_path}/sdw/models/interrogators\"#' {webui_path}/config.json\n",
         "!sed -i 's#\"additional_networks_extra_lora_path\": \".*models/Lora/\"#\"additional_networks_extra_lora_path\": \"{root_path}/sdw/models/Lora/\"#' {webui_path}/config.json\n",
+        "# ---\n",
+        "!sed -i 's/\"sd_checkpoint_hash\": \".*\"/\"sd_checkpoint_hash\": \"\"/g; s/\"sd_model_checkpoint\": \".*\"/\"sd_model_checkpoint\": \"\"/g; s/\"sd_vae\": \".*\"/\"sd_vae\": \"None\"/g' {webui_path}/config.json\n",
         "\n",
         "\n",
         "if env != \"SageMaker Studio Lab\":\n",

files_cells/notebooks/ru/downloading_ru.ipynb

@@ -80,6 +80,7 @@
         "            %conda update -q -n base conda\n",
         "            %conda install -q -y aria2\n",
         "            %conda install -q -y glib\n",
+        "            !pip install tensorflow\n",
         "\n",
         "            !wget -P /home/studio-lab-user https://huggingface.co/NagisaNao/fast_repo/resolve/main/sagemaker/FULL_DELETED_NOTEBOOK.ipynb\n",
         "\n",
@@ -160,6 +161,12 @@
         "    print(f\"⌚️ Вы проводите эту сессию в течение - \\033[33m{time_since_start}\\033[0m\")\n",
         "\n",
         "\n",
+        "# === ONLY SAGEMAKER FIXIN EXTENSIONS ===\n",
+        "if env == \"SageMaker Studio Lab\":\n",
+        "    # --- SuperMerger ---\n",
+        "    !wget -O {webui_path}/extensions/supermerger/scripts/mergers/mergers.py https://huggingface.co/NagisaNao/fast_repo/resolve/main/sagemaker/fixing/extensions/supermerger/scripts/mergers/mergers.py\n",
+        "\n",
+        "\n",
         "## Changes extensions and WebUi\n",
         "if latest_webui or latest_exstensions:\n",
         "    action = \"Обновление WebUI и Расширений\" if latest_webui and latest_exstensions else (\"Обновление WebUI\" if latest_webui else \"Обновление Расширений\")\n",
@@ -176,7 +183,10 @@
         "\n",
         "        ## Update extensions\n",
         "        if latest_exstensions:\n",
-        "            !{'for dir in ' + webui_path + '/extensions/*/; do cd \\\"$dir\\\" && git reset --hard && git pull; done'}\n",
+        "            if env != \"SageMaker Studio Lab\":\n",
+        "                !{'for dir in ' + webui_path + '/extensions/*/; do cd \\\"$dir\\\" && git reset --hard && git pull; done'}\n",
+        "            else:\n",
+        "                !{'for dir in /home/studio-lab-user/content/sdw/extensions/*/; do cd \\\"$dir\\\" && git fetch origin && git pull; done'}\n",
         "\n",
         "        !cd {webui_path}/repositories/stable-diffusion-stability-ai && git restore .\n",
         "    del cap\n",
@@ -463,7 +473,18 @@
         "!find {extensions_dir}/ -mindepth 1 -maxdepth 1 ! -name '*.txt' -printf '%f\\n'\n",
         "if any(not file.endswith(('.txt', '.yaml')) for file in os.listdir(control_dir)):\n",
         "    print(\"\\n\\033[33m➤ ControlNet\\033[0m\")\n",
-        "    !find {control_dir}/ -mindepth 1 ! -name '*.yaml' -printf '%f\\n' | sed 's/^[^_]*_[^_]*_[^_]*_\\(.*\\)_fp16\\.safetensors$/\\1/'"
+        "    !find {control_dir}/ -mindepth 1 ! -name '*.yaml' -printf '%f\\n' | sed 's/^[^_]*_[^_]*_[^_]*_\\(.*\\)_fp16\\.safetensors$/\\1/'\n",
+        "\n",
+        "\n",
+        "# === ONLY SAGEMAKER ===\n",
+        "if env == \"SageMaker Studio Lab\":\n",
+        "    with capture.capture_output() as cap:\n",
+        "        !rm -rf /home/studio-lab-user/.conda/envs/studiolab-safemode\n",
+        "        !rm -rf /home/studio-lab-user/.conda/envs/sagemaker-distribution\n",
+        "        !rm -rf /home/studio-lab-user/.conda/pkgs/cache\n",
+        "        !pip cache purge\n",
+        "        !rm -rf ~/.cache\n",
+        "    del cap"
       ],
       "metadata": {
         "id": "2lJmbqrs3Mu8"

files_cells/notebooks/ru/launch_ru.ipynb

@@ -87,6 +87,8 @@
         "# automatic fixing path V2\n",
         "!sed -i 's#\"tagger_hf_cache_dir\": \".*models/interrogators\"#\"tagger_hf_cache_dir\": \"{root_path}/sdw/models/interrogators\"#' {webui_path}/config.json\n",
         "!sed -i 's#\"additional_networks_extra_lora_path\": \".*models/Lora/\"#\"additional_networks_extra_lora_path\": \"{root_path}/sdw/models/Lora/\"#' {webui_path}/config.json\n",
+        "# ---\n",
+        "!sed -i 's/\"sd_checkpoint_hash\": \".*\"/\"sd_checkpoint_hash\": \"\"/g; s/\"sd_model_checkpoint\": \".*\"/\"sd_model_checkpoint\": \"\"/g; s/\"sd_vae\": \".*\"/\"sd_vae\": \"None\"/g' {webui_path}/config.json\n",
         "\n",
         "\n",
         "if env != \"SageMaker Studio Lab\":\n",

files_cells/python/en/downloading_en.py

@@ -67,6 +67,7 @@ if not os.path.exists(flag_file):
             get_ipython().run_line_magic('conda', 'update -q -n base conda')
             get_ipython().run_line_magic('conda', 'install -q -y aria2')
             get_ipython().run_line_magic('conda', 'install -q -y glib')
+            get_ipython().system('pip install tensorflow')
 
             get_ipython().system('wget -P /home/studio-lab-user https://huggingface.co/NagisaNao/fast_repo/resolve/main/sagemaker/FULL_DELETED_NOTEBOOK.ipynb')
 
@@ -147,6 +148,12 @@ else:
     print(f"⌚️ You have been conducting this session for - \033[33m{time_since_start}\033[0m")
 
 
+# === ONLY SAGEMAKER FIXIN EXTENSIONS ===
+if env == "SageMaker Studio Lab":
+    # --- SuperMerger ---
+    get_ipython().system('wget -O {webui_path}/extensions/supermerger/scripts/mergers/mergers.py https://huggingface.co/NagisaNao/fast_repo/resolve/main/sagemaker/fixing/extensions/supermerger/scripts/mergers/mergers.py')
+
+
 ## Changes extensions and WebUi
 if latest_webui or latest_exstensions:
     action = "Updating WebUI and Extensions" if latest_webui and latest_exstensions else ("WebUI Update" if latest_webui else "Update Extensions")
@@ -163,7 +170,10 @@ if latest_webui or latest_exstensions:
 
         ## Update extensions
         if latest_exstensions:
-            get_ipython().system('{\'for dir in \' + webui_path + \'/extensions/*/; do cd \\"$dir\\" && git reset --hard && git pull; done\'}')
+            if env != "SageMaker Studio Lab":
+                get_ipython().system('{\'for dir in \' + webui_path + \'/extensions/*/; do cd \\"$dir\\" && git reset --hard && git pull; done\'}')
+            else:
+                get_ipython().system('{\'for dir in /home/studio-lab-user/content/sdw/extensions/*/; do cd \\"$dir\\" && git fetch origin && git pull; done\'}')
 
         get_ipython().system('cd {webui_path}/repositories/stable-diffusion-stability-ai && git restore .')
     del cap
@@ -449,3 +459,14 @@ if any(not file.endswith(('.txt', '.yaml')) for file in os.listdir(control_dir))
     print("\n\033[33m➤ ControlNet\033[0m")
     get_ipython().system("find {control_dir}/ -mindepth 1 ! -name '*.yaml' -printf '%f\\n' | sed 's/^[^_]*_[^_]*_[^_]*_\\(.*\\)_fp16\\.safetensors$/\\1/'")
 
+
+# === ONLY SAGEMAKER ===
+if env == "SageMaker Studio Lab":
+    with capture.capture_output() as cap:
+        get_ipython().system('rm -rf /home/studio-lab-user/.conda/envs/studiolab-safemode')
+        get_ipython().system('rm -rf /home/studio-lab-user/.conda/envs/sagemaker-distribution')
+        get_ipython().system('rm -rf /home/studio-lab-user/.conda/pkgs/cache')
+        get_ipython().system('pip cache purge')
+        get_ipython().system('rm -rf ~/.cache')
+    del cap
+

files_cells/python/en/launch_en.py

@@ -69,6 +69,8 @@ if env != "SageMaker Studio Lab":
 # automatic fixing path V2
 get_ipython().system('sed -i \'s#"tagger_hf_cache_dir": ".*models/interrogators"#"tagger_hf_cache_dir": "{root_path}/sdw/models/interrogators"#\' {webui_path}/config.json')
 get_ipython().system('sed -i \'s#"additional_networks_extra_lora_path": ".*models/Lora/"#"additional_networks_extra_lora_path": "{root_path}/sdw/models/Lora/"#\' {webui_path}/config.json')
+# ---
+get_ipython().system('sed -i \'s/"sd_checkpoint_hash": ".*"/"sd_checkpoint_hash": ""/g; s/"sd_model_checkpoint": ".*"/"sd_model_checkpoint": ""/g; s/"sd_vae": ".*"/"sd_vae": "None"/g\' {webui_path}/config.json')
 
 
 if env != "SageMaker Studio Lab":

files_cells/python/ru/downloading_ru.py

@@ -67,6 +67,7 @@ if not os.path.exists(flag_file):
             get_ipython().run_line_magic('conda', 'update -q -n base conda')
             get_ipython().run_line_magic('conda', 'install -q -y aria2')
             get_ipython().run_line_magic('conda', 'install -q -y glib')
+            get_ipython().system('pip install tensorflow')
 
             get_ipython().system('wget -P /home/studio-lab-user https://huggingface.co/NagisaNao/fast_repo/resolve/main/sagemaker/FULL_DELETED_NOTEBOOK.ipynb')
 
@@ -147,6 +148,12 @@ else:
     print(f"⌚️ Вы проводите эту сессию в течение - \033[33m{time_since_start}\033[0m")
 
 
+# === ONLY SAGEMAKER FIXIN EXTENSIONS ===
+if env == "SageMaker Studio Lab":
+    # --- SuperMerger ---
+    get_ipython().system('wget -O {webui_path}/extensions/supermerger/scripts/mergers/mergers.py https://huggingface.co/NagisaNao/fast_repo/resolve/main/sagemaker/fixing/extensions/supermerger/scripts/mergers/mergers.py')
+
+
 ## Changes extensions and WebUi
 if latest_webui or latest_exstensions:
     action = "Обновление WebUI и Расширений" if latest_webui and latest_exstensions else ("Обновление WebUI" if latest_webui else "Обновление Расширений")
@@ -163,7 +170,10 @@ if latest_webui or latest_exstensions:
 
         ## Update extensions
         if latest_exstensions:
-            get_ipython().system('{\'for dir in \' + webui_path + \'/extensions/*/; do cd \\"$dir\\" && git reset --hard && git pull; done\'}')
+            if env != "SageMaker Studio Lab":
+                get_ipython().system('{\'for dir in \' + webui_path + \'/extensions/*/; do cd \\"$dir\\" && git reset --hard && git pull; done\'}')
+            else:
+                get_ipython().system('{\'for dir in /home/studio-lab-user/content/sdw/extensions/*/; do cd \\"$dir\\" && git fetch origin && git pull; done\'}')
 
         get_ipython().system('cd {webui_path}/repositories/stable-diffusion-stability-ai && git restore .')
     del cap
@@ -449,3 +459,14 @@ if any(not file.endswith(('.txt', '.yaml')) for file in os.listdir(control_dir))
     print("\n\033[33m➤ ControlNet\033[0m")
     get_ipython().system("find {control_dir}/ -mindepth 1 ! -name '*.yaml' -printf '%f\\n' | sed 's/^[^_]*_[^_]*_[^_]*_\\(.*\\)_fp16\\.safetensors$/\\1/'")
 
+
+# === ONLY SAGEMAKER ===
+if env == "SageMaker Studio Lab":
+    with capture.capture_output() as cap:
+        get_ipython().system('rm -rf /home/studio-lab-user/.conda/envs/studiolab-safemode')
+        get_ipython().system('rm -rf /home/studio-lab-user/.conda/envs/sagemaker-distribution')
+        get_ipython().system('rm -rf /home/studio-lab-user/.conda/pkgs/cache')
+        get_ipython().system('pip cache purge')
+        get_ipython().system('rm -rf ~/.cache')
+    del cap
+

files_cells/python/ru/launch_ru.py

@@ -69,6 +69,8 @@ if env != "SageMaker Studio Lab":
 # automatic fixing path V2
 get_ipython().system('sed -i \'s#"tagger_hf_cache_dir": ".*models/interrogators"#"tagger_hf_cache_dir": "{root_path}/sdw/models/interrogators"#\' {webui_path}/config.json')
 get_ipython().system('sed -i \'s#"additional_networks_extra_lora_path": ".*models/Lora/"#"additional_networks_extra_lora_path": "{root_path}/sdw/models/Lora/"#\' {webui_path}/config.json')
+# ---
+get_ipython().system('sed -i \'s/"sd_checkpoint_hash": ".*"/"sd_checkpoint_hash": ""/g; s/"sd_model_checkpoint": ".*"/"sd_model_checkpoint": ""/g; s/"sd_vae": ".*"/"sd_vae": "None"/g\' {webui_path}/config.json')
 
 
 if env != "SageMaker Studio Lab":