zootkitty commited on
Commit
5358099
·
verified ·
1 Parent(s): 62a7fb7

conversion script

Browse files
Files changed (1) hide show
  1. convert_model.py +13 -0
convert_model.py ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from safetensors.torch import load_file, save_file
2
+
3
+ source_model = load_file("Chroma1-HD.safetensors")
4
+ target_model = load_file("gonzalomoChroma_v12FP8UNET.safetensors")
5
+ converted_model = {}
6
+ for key in source_model:
7
+ if key not in target_model:
8
+ converted_model[key] = source_model[key]
9
+ for key in target_model:
10
+ if key in source_model:
11
+ converted_model[key] = target_model[key]
12
+
13
+ save_file(converted_model, "gonzalomoChroma_v12FP8UNET_converted.safetensors")