Upload reconstruct_model.py with huggingface_hub

reconstruct_model.py

@@ -0,0 +1,43 @@
+#!/usr/bin/env python3
+"""
+Script to reconstruct original files from split parts.
+Run this after downloading the model to reconstruct large files.
+"""
+
+import json
+import os
+from pathlib import Path
+
+def reconstruct_files():
+    """Reconstruct original files from parts using the manifest."""
+    with open("split_manifest.json", 'r') as f:
+        manifest = json.load(f)
+    
+    for original_file, info in manifest.items():
+        print(f"Reconstructing {original_file}...")
+        
+        output_path = Path(original_file)
+        output_path.parent.mkdir(parents=True, exist_ok=True)
+        
+        with open(output_path, 'wb') as output_file:
+            for part_file in info["parts"]:
+                with open(part_file, 'rb') as part:
+                    output_file.write(part.read())
+        
+        # Verify size
+        if output_path.stat().st_size == info["original_size"]:
+            print(f"  ✓ Reconstructed successfully ({output_path.stat().st_size} bytes)")
+            
+            # Clean up part files
+            for part_file in info["parts"]:
+                os.remove(part_file)
+                print(f"  Removed part: {part_file}")
+        else:
+            print(f"  ✗ Size mismatch for {original_file}")
+    
+    # Remove manifest after successful reconstruction
+    os.remove("split_manifest.json")
+    print("Reconstruction complete!")
+
+if __name__ == "__main__":
+    reconstruct_files()