Update sh.py
Browse files
sh.py
CHANGED
@@ -4,8 +4,8 @@ from safetensors.torch import load_file
|
|
4 |
|
5 |
|
6 |
# Load both models
|
7 |
-
model1 = load_file('
|
8 |
-
model2 = load_file('diffusion_pytorch_model-
|
9 |
|
10 |
# Iterate through the tensor names and shapes
|
11 |
for name in model1.keys():
|
|
|
4 |
|
5 |
|
6 |
# Load both models
|
7 |
+
model1 = load_file('merged_model_016.safetensors')
|
8 |
+
model2 = load_file('diffusion_pytorch_model-00001-of-00003.safetensors')
|
9 |
|
10 |
# Iterate through the tensor names and shapes
|
11 |
for name in model1.keys():
|