Views
No views yet
1import sd_mecha
2sd_mecha.set_log_level()
3
4
5text_encoder_recipe = sd_mecha.model("noobaiXLNAIXL_vPred10Version.safetensors", "sdxl")
6
7unet_recipe = sd_mecha.add_perpendicular(
8 sd_mecha.model("noobaiXLNAIXL_vPred10Version.safetensors", "sdxl"),
9 sd_mecha.model("cyberrealisticXL_v4.safetensors", "sdxl"),
10 sd_mecha.model("sd_xl_base_1.0_0.9vae.safetensors", "sdxl"),
11)
12
13recipe = sd_mecha.weighted_sum(
14 text_encoder_recipe,
15 unet_recipe,
16 alpha=(
17 sd_mecha.blocks("sdxl", "txt") |
18 sd_mecha.blocks("sdxl", "txt2") |
19 sd_mecha.default("sdxl", "unet", 1)
20 ),
21)
22
23merger = sd_mecha.RecipeMerger(
24 models_dir=r"C:\path\to\models\directory",
25)
26
27merger.merge_and_save(recipe, output="output.safetensors")1from safetensors.torch import load_file, save_file
2import torch
3
4state_dict = load_file("output.safetensors")
5state_dict["v_pred"] = torch.tensor([])
6state_dict["ztsnr"] = torch.tensor([])
7save_file(state_dict, "noobaiXLNAIXL_vPred10Version-cyberrealistic4-perpendicular.safetensors")