Update app.py

app.py

@@ -1,112 +1,14 @@
-import argparse
-import platform
-import subprocess
-
-
-def pip_install_packages(packages, extra_index_url=None, verbose=False, pre=False):
-    for package in packages:
-        try:
-            print(f"..installing {package}")
-            
-            # base command
-            cmd = ["pip", "install"]
-
-            if pre:
-                cmd.append("--pre")
-
-            # add '-q' if not verbose
-            if not verbose:
-                cmd.append("-q")
-
-            # add package name
-            cmd.append(package)
-
-            # add extra_index_url if it exists
-            if extra_index_url:
-                cmd.extend(["--extra-index-url", extra_index_url])
-
-            if verbose:
-                print(cmd)
-
-            # run the command and capture output
-            result = subprocess.run(cmd, capture_output=not verbose, text=True)
-            
-            if verbose:
-                # print stdout and stderr if verbose
-                print(result.stdout)
-                print(result.stderr)
-
-        except Exception as e:
-            print(f"failed to install {package}: {e}")
-    return
-
-
-def install_requirements(verbose=False):
-
-    # Detect System
-    os_system = platform.system()
-    print(f"system detected: {os_system}")
-
-    # Install pytorch
-    torch = [
-        "torch",
-        "torchvision",
-        "torchaudio"
-    ]
-    extra_index_url = "https://download.pytorch.org/whl/nightly/cu121"
-    pip_install_packages(torch, extra_index_url=extra_index_url, verbose=verbose, pre=True)
-
-    # List of common packages to install
-    common = [
-        "clean-fid",
-        "colab-convert",
-        "einops",
-        "ftfy",
-        "ipython",
-        "ipywidgets",
-        "jsonmerge",
-        "jupyterlab",
-        "jupyter_http_over_ws",
-        "kornia",
-        "matplotlib",
-        "notebook",
-        "numexpr",
-        "omegaconf",
-        "opencv-python",
-        "pandas",
-        "pytorch_lightning==1.7.7",
-        "resize-right",
-        "scikit-image==0.19.3",
-        "scikit-learn",
-        "timm",
-        "torchdiffeq",
-        "transformers==4.19.2",
-        "safetensors",
-        "albumentations",
-        "more_itertools",
-        "devtools",
-        "validators",
-        "numpngw",
-        "open-clip-torch",
-        "torchsde",
-        "ninja",
-        "pydantic",
-    ]
-    pip_install_packages(common)
-
-    # Xformers install
-    linux_xformers = [
-        "triton",
-        "xformers==0.0.21.dev546",
-    ]
-    windows_xformers = [
-        "xformers==0.0.21.dev546",
-    ]
-    xformers = windows_xformers if os_system == 'Windows' else linux_xformers
-    pip_install_packages(xformers)
-
-if __name__ == "__main__":
-    parser = argparse.ArgumentParser()
-    parser.add_argument('--verbose', action='store_true', help='print pip install stuff')
-    args = parser.parse_args()
-    install_requirements(verbose=args.verbose)
+import os
+os.chdir(f"/home/users/app")
+os.system(f"apt -y update -qq")
+os.system(f"wget https://github.com/camenduru/gperftools/releases/download/v1.0/libtcmalloc_minimal.so.4 -O /home/users/app/libtcmalloc_minimal.so.4")
+os.system(f"env LD_PRELOAD=/content/libtcmalloc_minimal.so.4")
+os.system(f"apt -y install -qq aria2")
+os.system(f"pip install -q xformers==0.0.20 triton==2.0.0 -U")
+os.system(f"git clone -b v2.6 https://dagshub.com/camenduru/ui")
+os.system(f"git clone https://github.com/camenduru/control /home/users/app/ui/extensions/control")
+os.system(f"cd /home/users/app/ui")
+os.system(f"git reset --hard")
+os.system(f"git -C /content/ui/repositories/stable-diffusion-stability-ai reset --hard")
+os.system(f"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/sd_xl_base_1.0/resolve/main/sd_xl_base_1.0_0.9vae.safetensors -d /home/users/app/ui/models/Stable-diffusion -o sd_xl_base_1.0.safetensors")
+os.system(f"python launch.py --listen --xformers --enable-insecure-extension-access --theme dark --gradio-queue --multiple")