import torch
import safetensors
import sd_mecha
sd_mecha.set_log_level()

recipe = sd_mecha.rotate(
    sd_mecha.model("smoothftspo.safetensors", "sdxl"),
    sd_mecha.model("v14t1s.safetensors", "sdxl"),
)

merger = sd_mecha.RecipeMerger(
    models_dir=r"output_dir",
)

merger.merge_and_save(recipe, output="result.safetensors")

#
# (repeated, adding the result from the first one; rotate with it in the middle)
#

import torch
import safetensors
import sd_mecha
sd_mecha.set_log_level()

recipe = sd_mecha.rotate(
    sd_mecha.model("smoothftspo.safetensors", "sdxl"),
    sd_mecha.model("result.safetensors", "sdxl"),
    sd_mecha.model("v14t1s.safetensors", "sdxl"),
)

merger = sd_mecha.RecipeMerger(
    models_dir=r"output_dir",
)

merger.merge_and_save(recipe, output="final.safetensors")