Update co.py
Browse files
co.py
CHANGED
@@ -2,9 +2,9 @@ import torch
|
|
2 |
from safetensors.torch import load_file, save_file
|
3 |
|
4 |
# Load all three models
|
5 |
-
model_1 = load_file("
|
6 |
-
model_2 = load_file("
|
7 |
-
model_3 = load_file("
|
8 |
|
9 |
# Combine the models into a single dictionary
|
10 |
combined_model = {**model_1, **model_2, **model_3}
|
|
|
2 |
from safetensors.torch import load_file, save_file
|
3 |
|
4 |
# Load all three models
|
5 |
+
model_1 = load_file("merged_model_08.safetensors")
|
6 |
+
model_2 = load_file("merged_model_8.safetensors")
|
7 |
+
model_3 = load_file("merged_model_28.safetensors")
|
8 |
|
9 |
# Combine the models into a single dictionary
|
10 |
combined_model = {**model_1, **model_2, **model_3}
|