launch.py

import os
import sys
import platform
import version
import warnings
from pathlib import Path
import ssl
import argparse

ssl._create_default_https_context = ssl._create_unverified_context

warnings.filterwarnings("ignore", category=FutureWarning, module="insightface")
warnings.filterwarnings("ignore", category=FutureWarning, module="transformers")
warnings.filterwarnings("ignore", category=UserWarning, module="torchvision")
warnings.filterwarnings("ignore", category=UserWarning, module="gradio")
warnings.filterwarnings("ignore", category=UserWarning, module="torchsde")
warnings.filterwarnings("ignore", category=UserWarning)

warnings.filterwarnings(
    "ignore", category=UserWarning, module="torchvision.transforms.functional_tensor"
)
warnings.filterwarnings(
    "ignore", category=UserWarning, message="TypedStorage is deprecated"
)

from modules.launch_util import (
    is_installed,
    run,
    python,
    run_pip,
    repo_dir,
    requirements_met,
    script_path,
    dir_repos,
)

torch_index_url = "https://download.pytorch.org/whl/cu121"
requirements_file = "requirements_versions.txt"
launch_pip_file = "pip_modules.txt"

git_repos = [
    {
        "name": "ComfyUI",
        "path": "ComfyUI",
        "url": "https://github.com/comfyanonymous/ComfyUI",
        "hash": "935ae153e154813ace36db4c4656a5e96f403eba",
        "add_path": "ComfyUI",
    },
    {
        "name": "Stable Fast 3D",
        "path": "stable-fast-3d",
        "url": "https://github.com/Stability-AI/stable-fast-3d.git",
        "hash": "070ece138459e38e1fe9f54aa19edb834bced85e",
        "add_path": "stable-fast-3d",
    },
    {
        "name": "ComfyUI-GGUF",
        "path": "comfyui_gguf",
        "url": "https://github.com/city96/ComfyUI-GGUF.git",
        "hash": "6561064dcfb3dfa638e3739506acfd34924e1cc5",
        "add_path": "",
    },
]

REINSTALL_ALL = False
if os.path.exists("reinstall"):
    REINSTALL_ALL = True

def prepare_environment():
    print(f"Python {sys.version}")
    print(f"RuinedFooocus version: {version.version}")

    run(f'"{python}" -m pip install --upgrade pip', "Check pip", "Couldn't check pip", live=False)
    run(f'"{python}" -m pip install -r "{requirements_file}"', "Check pre-requirements", "Couldn't check pre-reqs", live=False)

    if REINSTALL_ALL or not requirements_met(launch_pip_file):
        print("This next step may take a while")
        os.environ["FLASH_ATTENTION_SKIP_CUDA_BUILD"] = "TRUE"
        run_pip(f'install -r "{launch_pip_file}" --extra-index-url {torch_index_url}', "required modules")

def clone_git_repos():
    from modules.launch_util import git_clone

    for repo in git_repos:
        git_clone(repo["url"], repo_dir(repo["path"]), repo["name"], repo["hash"])
        add_path = str(Path(script_path) / dir_repos / repo["add_path"])
        if add_path not in sys.path:
            sys.path.append(add_path)

def download_models():
    from modules.util import load_file_from_url
    from shared import path_manager

    model_filenames = [
        (
            path_manager.model_paths["modelfile_path"],
            "sd_xl_base_1.0_0.9vae.safetensors",
            "https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/resolve/main/sd_xl_base_1.0_0.9vae.safetensors",
        ),
        (
            path_manager.model_paths["lorafile_path"],
            "sd_xl_offset_example-lora_1.0.safetensors",
            "https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/resolve/main/sd_xl_offset_example-lora_1.0.safetensors",
        ),
        (
            path_manager.model_paths["lorafile_path"],
            "lcm-lora-sdxl.safetensors",
            "https://huggingface.co/latent-consistency/lcm-lora-sdxl/resolve/main/pytorch_lora_weights.safetensors",
        ),
        (
            path_manager.model_paths["lorafile_path"],
            "lcm-lora-ssd-1b.safetensors",
            "https://huggingface.co/latent-consistency/lcm-lora-ssd-1b/resolve/main/pytorch_lora_weights.safetensors",
        ),
        (
            path_manager.model_paths["vae_approx_path"],
            "taesdxl_decoder",
            "https://github.com/madebyollin/taesd/raw/main/taesdxl_decoder.pth",
        ),
        (
            "prompt_expansion",
            "pytorch_model.bin",
            "https://huggingface.co/lllyasviel/misc/resolve/main/fooocus_expansion.bin",
        ),
        (
            "models/layerdiffuse/",
            "layer_xl_transparent_attn.safetensors",
            "https://huggingface.co/LayerDiffusion/layerdiffusion-v1/resolve/main/layer_xl_transparent_attn.safetensors",
        ),
        (
            "models/layerdiffuse/",
            "vae_transparent_decoder.safetensors",
            "https://huggingface.co/LayerDiffusion/layerdiffusion-v1/resolve/main/vae_transparent_decoder.safetensors",
        ),
    ]

    for model_dir, file_name, url in model_filenames:
        load_file_from_url(
            url=url,
            model_dir=model_dir,
            file_name=file_name,
        )

from argparser import args

if args.gpu_device_id is not None:
    os.environ['CUDA_VISIBLE_DEVICES'] = str(args.gpu_device_id)
    print("Set device to:", args.gpu_device_id)

prepare_environment()
if os.path.exists("reinstall"):
    os.remove("reinstall")

try:
    clone_git_repos()
except:
    print(f"WARNING: Failed checking git-repos. Trying to start without update.")

download_models()

from webui import *