Update rp.py

rp.py

@@ -1,7 +1,7 @@
 import torch
 from safetensors.torch import load_file, save_file
 
-def reduce_key_size(input_file, output_file, reduction_factor=0.50):
+def reduce_key_size(input_file, output_file, reduction_factor=0.45):
     # Load the model
     model_data = load_file(input_file)
 
@@ -12,18 +12,15 @@ def reduce_key_size(input_file, output_file, reduction_factor=0.50):
         # Calculate the new size
         new_size = int(original_tensor.size(0) * (1 - reduction_factor))
         
-        # Resize the tensor (ensure the new size is positive)
-        if new_size > 0:
+        # Resize the tensor (this could vary depending on your requirements)
+        if new_size > 0:  # Ensure new size is positive
             reduced_tensor = original_tensor[:new_size]
-            
-            # Convert to FP16 precision (half-precision floating point)
-            fp16_tensor = reduced_tensor.to(torch.float16)
-            model_data[key] = fp16_tensor
+            model_data[key] = reduced_tensor
     
     # Save the modified model
     save_file(model_data, output_file)
 
 # Usage example
-input_file = 'merged_model_216.safetensors'  # Replace with your input model file
-output_file = 'merged_model_28.safetensors'  # Desired output file name
+input_file = 'merged_model_08.safetensors'  # Replace with your input model file
+output_file = 'merged_model_06.safetensors'  # Desired output file name
 reduce_key_size(input_file, output_file)