reoe's picture
Create recipe.py
100dea1 verified
raw
history blame
799 Bytes
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")