import sd_mecha
sd_mecha.set_log_level()


text_encoder_recipe = sd_mecha.model("NoobAI-XL-Vpred-v1.0+v29b-v2.safetensors", "sdxl")

unet_recipe = sd_mecha.add_perpendicular(
    sd_mecha.model("NoobAI-XL-Vpred-v1.0+v29b-v2.safetensors", "sdxl"),
    sd_mecha.model("cyberrealisticXL_v4.safetensors", "sdxl"),
    sd_mecha.model("sd_xl_base_1.0.safetensors", "sdxl"),
)

recipe = sd_mecha.weighted_sum(
    text_encoder_recipe,
    unet_recipe,
    alpha=(
        sd_mecha.blocks("sdxl", "txt") |
        sd_mecha.blocks("sdxl", "txt2") |
        sd_mecha.default("sdxl", "unet", 1)
    ),
)

merger = sd_mecha.RecipeMerger(
    models_dir=r"D:\ComfyUI_windows_portable\ComfyUI\models\checkpoints",
)

merger.merge_and_save(recipe, output="noobV10+v29b-cyberfix-perpendicular.safetensors")