File size: 479 Bytes
5358099
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
from safetensors.torch import load_file, save_file

source_model = load_file("Chroma1-HD.safetensors")
target_model = load_file("gonzalomoChroma_v12FP8UNET.safetensors")
converted_model = {}
for key in source_model:
    if key not in target_model:
        converted_model[key] = source_model[key]
for key in target_model:
    if key in source_model:
        converted_model[key] = target_model[key]

save_file(converted_model, "gonzalomoChroma_v12FP8UNET_converted.safetensors")