Merge pull request #13276 from woweenie/patch-1

import torch

import re

import safetensors.torch

from omegaconf import OmegaConf

from omegaconf import OmegaConf, ListConfig

from os import mkdir

from urllib import request

import ldm.modules.midas as midas

from modules import paths, shared, modelloader, devices, script_callbacks, sd_vae, sd_disable_initialization, errors, hashes, sd_models_config, sd_unet, sd_models_xl, cache, extra_networks, processing, lowvram, sd_hijack

from modules.timer import Timer

import tomesd

import numpy as np

model_dir = "Stable-diffusion"

model_path = os.path.abspath(os.path.join(paths.models_path, model_dir))

    os.makedirs(model_path, exist_ok=True)

    enable_midas_autodownload()

    patch_given_betas()

def checkpoint_tiles(use_short=False):

    midas.api.load_model = load_model_wrapper

def patch_given_betas():

    original_register_schedule = ldm.models.diffusion.ddpm.DDPM.register_schedule

    def patched_register_schedule(*args, **kwargs):

        if args[1] is not None and isinstance(args[1], ListConfig):

            modified_args = list(args)  # Convert args tuple to a list

            modified_args[1] = np.array(args[1])  # Modify the desired element

            args = tuple(modified_args)  # Convert the list back to a tuple

        original_register_schedule(*args, **kwargs)

    ldm.models.diffusion.ddpm.DDPM.register_schedule = patched_register_schedule

def repair_config(sd_config):

    if not hasattr(sd_config.model.params, "use_ema"):