conversion script
Browse files- 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")
|