Upload folder using huggingface_hub

README.md

@@ -5,3 +5,5 @@ This repository contains the final trained model and intermediate checkpoints.
 - The main directory contains the fully trained model (checkpoint 0).
 - The `checkpoints` directory contains all intermediate checkpoints.
 
+Now updated to match tetrahedron format
+

checkpoints/checkpoint-100.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8ff7f82119e283431e20ac67a435fc12057b0eea0bab2394b03a5b579181ebd7
-size 3252234
+oid sha256:bdddf11f6c354f0fe93e9e0304366e3c753926157bc0ef8d6b726d850adcd139
+size 2480362

checkpoints/checkpoint-25.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:03c5f53e08a4c2801e6c15b60c4838fecfb00c65b15a114f48484f10bca5219d
-size 3252119
+oid sha256:b47a7288186dd7e81fc0b4dff55ea628b4305471cf642159bb5fbefdfdc6c927
+size 2480328

checkpoints/checkpoint-50.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b2bfdfd1f19ac378a3558705f82912da77eadb4730893d62d8eb6d4345602158
-size 3252119
+oid sha256:404c5ce1e19ed45746ed0d35635e4f1081dd1917fb12cae1b1a63ca7208bd0c7
+size 2480328

checkpoints/checkpoint-75.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:891f63bac1d31784a5978a8749d549b580ef6ba9027f2df95db216bbd0733a92
-size 3252119
+oid sha256:3beb8688ec66697e58f601b78412a8d38a05ff7f8fbf8cb0f9e91a2e2a51989e
+size 2480328

convert_checkpoints.py

@@ -0,0 +1,37 @@
+# convert_and_replace.py
+import os
+from pathlib import Path
+
+import torch
+
+
+def convert_checkpoint_format(checkpoint_path):
+    """Convert nested format to direct format in place."""
+    # Load the checkpoint
+    checkpoint = torch.load(checkpoint_path, map_location='cpu')
+    
+    # Extract just the model state dict
+    if 'model' in checkpoint:
+        model_state_dict = checkpoint['model']
+        print(f"Converting {checkpoint_path}: nested -> direct format")
+        
+        # Save back in direct format
+        torch.save(model_state_dict, checkpoint_path)
+        print(f"Updated: {checkpoint_path}")
+    else:
+        print(f"Already in direct format: {checkpoint_path}")
+
+def convert_all_checkpoints():
+    """Convert all checkpoint files in the current directory."""
+    checkpoints_dir = Path("checkpoints")
+    
+    if not checkpoints_dir.exists():
+        print("No checkpoints directory found")
+        return
+    
+    # Convert all .pt files
+    for checkpoint_file in checkpoints_dir.glob("*.pt"):
+        convert_checkpoint_format(checkpoint_file)
+
+if __name__ == "__main__":
+    convert_all_checkpoints()