Create fix.py

fix.py

@@ -0,0 +1,9 @@
+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: Print the size (shape) of each tensor
+for tensor_name, tensor in checkpoint.items():
+    print(f"Tensor Name: {tensor_name}, Size: {tensor.shape}")