Rename fix.py to sh.py

fix.py

@@ -1,13 +0,0 @@
-from safetensors.torch import load_file
-
-# Step 1: Load the safetensors file
-checkpoint_path = 'flowgram.safetensors'  # Replace with your actual file path
-checkpoint = load_file(checkpoint_path)
-
-# Step 2: Open a log file to save the output
-with open("log.txt", "w") as log_file:
-    # Step 3: Write the size (shape) of each tensor to the file
-    for tensor_name, tensor in checkpoint.items():
-        log_file.write(f"Tensor Name: {tensor_name}, Size: {tensor.shape}\n")
-
-print("Tensor sizes saved to log.txt")

sh.py

@@ -0,0 +1,19 @@
+import safetensors
+
+# Load both models
+model1 = safetensors.load_file('merged_model0.safetensors')
+model2 = safetensors.load_file('diffusion_pytorch_model-00001-of-00003.safetensors')
+
+# Iterate through the tensor names and shapes
+for name in model1.keys():
+    if name in model2:
+        shape1 = model1[name].shape
+        shape2 = model2[name].shape
+        if shape1 != shape2:
+            print(f"Tensor '{name}' has different shapes: Model 1: {shape1}, Model 2: {shape2}")
+    else:
+        print(f"Tensor '{name}' is not present in model 2.")
+
+for name in model2.keys():
+    if name not in model1:
+        print(f"Tensor '{name}' is not present in model 1.")