reoe commited on
Commit
100dea1
1 Parent(s): 48eddec

Create recipe.py

Browse files
Files changed (1) hide show
  1. recipe.py +36 -0
recipe.py ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ import safetensors
3
+ import sd_mecha
4
+ sd_mecha.set_log_level()
5
+
6
+ recipe = sd_mecha.rotate(
7
+ sd_mecha.model("smoothftspo.safetensors", "sdxl"),
8
+ sd_mecha.model("v14t1s.safetensors", "sdxl"),
9
+ )
10
+
11
+ merger = sd_mecha.RecipeMerger(
12
+ models_dir=r"output_dir",
13
+ )
14
+
15
+ merger.merge_and_save(recipe, output="result.safetensors")
16
+
17
+ #
18
+ # (repeated, adding the result from the first one; rotate with it in the middle)
19
+ #
20
+
21
+ import torch
22
+ import safetensors
23
+ import sd_mecha
24
+ sd_mecha.set_log_level()
25
+
26
+ recipe = sd_mecha.rotate(
27
+ sd_mecha.model("smoothftspo.safetensors", "sdxl"),
28
+ sd_mecha.model("result.safetensors", "sdxl"),
29
+ sd_mecha.model("v14t1s.safetensors", "sdxl"),
30
+ )
31
+
32
+ merger = sd_mecha.RecipeMerger(
33
+ models_dir=r"output_dir",
34
+ )
35
+
36
+ merger.merge_and_save(recipe, output="final.safetensors")