Upload folder using huggingface_hub

checkpoints/checkpoint-100.pt

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

checkpoints/checkpoint-25.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b47a7288186dd7e81fc0b4dff55ea628b4305471cf642159bb5fbefdfdc6c927
-size 2480328
+oid sha256:200f2670bbfd05c752c6af8e5e3835b47ff11e4c7286203f0bbfaa0c6fbead11
+size 2478049

checkpoints/checkpoint-50.pt

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

checkpoints/checkpoint-75.pt

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

convert_checkpoints.py

@@ -10,16 +10,27 @@ def convert_checkpoint_format(checkpoint_path):
     # 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}")
+
+    model_state_dict = checkpoint
+    print(f"Converting {checkpoint_path}: nested -> direct format")
+    
+    # Create new state dict with flattened keys
+    new_state_dict = {}
+    for key, value in model_state_dict.items():
+        if key.startswith('model.'):
+            # Remove 'model.' prefix
+            new_key = key[6:]  # Remove 'model.' (6 characters)
+            print(f"Converting key: '{key}' -> '{new_key}'")
+            new_state_dict[new_key] = value
+        else:
+            # Keep keys that don't start with 'model.'
+            print(f"Keeping key: '{key}'")
+            new_state_dict[key] = value
+    
+    # Save back in direct format
+    torch.save(new_state_dict, checkpoint_path)
+    print(f"Updated: {checkpoint_path}")
+
 
 def convert_all_checkpoints():
     """Convert all checkpoint files in the current directory."""