Update setup_forge.py

setup_forge.py

@@ -5,6 +5,7 @@ from pathlib import Path
 import subprocess, time, os, shlex
 from nenen88 import pull, say, download, clone, tempe
 
+
 xxx = "/home/studio-lab-user"
 zzz = Path(xxx) / "forge"
 
@@ -16,19 +17,20 @@ if zzz.exists():
 
         get_ipython().system("git pull origin master")
         get_ipython().system("git fetch --tags")
-    
+
 else:
     mama = f"{xxx}/.conda/setup.css"
     fff = {"stdout": subprocess.DEVNULL, "stderr": subprocess.DEVNULL}
+
     get_ipython().system(f"rm -rf {xxx}/.zrok")
     get_ipython().system(f"curl -sLo {mama} https://github.com/gutris1/segsmaker/raw/main/ui/sd/asd/setup.css")
-
+    
     gariz = """<div class="gradient-text">asd</div>"""
     garis = widgets.Output()
     b1 = widgets.Button(description='SD 1.5')
     b2 = widgets.Button(description='SDXL')
     ikuuuhhh = widgets.Button(description='Install')
-    
+
     b1b2 = widgets.HBox([b1, b2], layout=Layout(justify_content='space-between'))
     sd_setup = widgets.Output()
     boxxx = widgets.VBox([b1b2, ikuuuhhh], layout=Layout(
@@ -46,12 +48,14 @@ else:
     boxxx.add_class("boxxx")
     selected = [None]
 
+
     def ccss(mama):
         with open(mama, "r") as gantung:
             susu = gantung.read()
 
         display(HTML(f"<style>{susu}</style>"))
 
+
     def req_list(xxx, zzz):
         return [
             f"pip install -q -r requirements.txt basicsr insightface onnxruntime-gpu",
@@ -63,7 +67,9 @@ else:
             f"ln -vs /tmp/lora {zzz}/models/Lora/tmp_lora",
             f"ln -vs /tmp/z123 {zzz}/models/z123",
             f"ln -vs /tmp/svd {zzz}/models/svd",
-            f"ln -vs /tmp/controlnet {zzz}/models/ControlNet"]
+            f"ln -vs /tmp/controlnet {zzz}/models/ControlNet"
+        ]
+
 
     def sd_clone(xxx, zzz, fff):
         time.sleep(1)
@@ -75,10 +81,17 @@ else:
             subprocess.run(shlex.split(lines), **fff)
 
 
+    def ext(zzz):
+        say("<br><b>【{red} Installing Extensions{d} 】{red}</b>")
+        os.chdir(f"{zzz}/extensions")
+        clone(f"{zzz}/asd/extension.txt")
+        
+
     def sd_1_5(xxx, zzz, fff):
         sd_clone(xxx, zzz, fff)
 
         extras = [
+            f"https://huggingface.co/pantat88/ui/resolve/main/embeddings.zip {zzz}",
             f"https://civitai.com/api/download/models/150491 {zzz}/embeddings edgQuality.pt",
             f"https://huggingface.co/pantat88/ui/resolve/main/4x-UltraSharp.pth {zzz}/models/ESRGAN",
             f"https://huggingface.co/pantat88/ui/resolve/main/4x-AnimeSharp.pth {zzz}/models/ESRGAN",
@@ -94,6 +107,7 @@ else:
 
         for items in extras:
             download(items)
+
         get_ipython().system(f"curl -sLo {zzz}/asd/modelos.py https://huggingface.co/Carlos2312/asd/resolve/main/modelos.py")
         get_ipython().system(f"rm -rf {zzz}/asd/extension.txt")