sd-mecha 0.0.2

Stable Diffusion recipe merger

sd-mecha

sd-mecha is a stable diffusion recipe merger:

import sd_mecha

# create a simple weighted sum recipe
recipe = sd_mecha.weighted_sum(
    sd_mecha.weighted_sum(
        "ghostmix_v20Bakedvae",
        "deliberate_v2",
        alpha=0.5,
    ),
    "dreamshaper_332BakedVaeClipFix",
    alpha=0.33,
)

# scheduler contains default parameters
scheduler = sd_mecha.MergeScheduler(
    base_dir=r"E:\sd\models\Stable-diffusion",
)

# perform the entire merge plan and save to output path
scheduler.merge_and_save(recipe, output_path="basic_merge")

See the examples directory for other examples.

Motivation

Keeping track of full merge recipes has always been annoying. I needed something that allows to store merge recipes in a readable format while also being executable. I also needed something that allows to fully merge an entire tree of models without having to save intermediate models to disk.

Typically, mergers load all models in memory before initiating the merge process. This can be very inefficient when the merge focuses on each key individually:

sd-mecha doesn't have this problem as it saves keys as soon as it can:

This allows to merge a very large number of models simultaneously on low-end hardware. (i.e. 8+)

Install

pip install sd-mecha torch

sd-mecha depends additionally on:

• torch>=2.0.1

The pypi package does not ship with torch so that you can install the appropriate version for your system.