Upload combine.py

quantized/combine.py

@@ -0,0 +1,11 @@
+import safetensors.torch  
+  
+# Load both files  
+quantized_blocks = safetensors.torch.load_file("transformer_blocks.safetensors")  
+unquantized_layers = safetensors.torch.load_file("unquantized_layers.safetensors")  
+  
+# Combine the state dicts  
+combined_state_dict = {**quantized_blocks, **unquantized_layers}  
+  
+# Save the combined model  
+safetensors.torch.save_file(combined_state_dict, "combined.safetensors")