pranavajay commited on
Commit
2686611
·
verified ·
1 Parent(s): 8bf1132

Update rp.py

Browse files
Files changed (1) hide show
  1. rp.py +3 -3
rp.py CHANGED
@@ -1,7 +1,7 @@
1
  import torch
2
  from safetensors.torch import load_file, save_file
3
 
4
- def reduce_key_size(input_file, output_file, reduction_factor=0.45):
5
  # Load the model
6
  model_data = load_file(input_file)
7
 
@@ -21,6 +21,6 @@ def reduce_key_size(input_file, output_file, reduction_factor=0.45):
21
  save_file(model_data, output_file)
22
 
23
  # Usage example
24
- input_file = 'merged_model_28.safetensors' # Replace with your input model file
25
- output_file = 'merged_model_26.safetensors' # Desired output file name
26
  reduce_key_size(input_file, output_file)
 
1
  import torch
2
  from safetensors.torch import load_file, save_file
3
 
4
+ def reduce_key_size(input_file, output_file, reduction_factor=0.50):
5
  # Load the model
6
  model_data = load_file(input_file)
7
 
 
21
  save_file(model_data, output_file)
22
 
23
  # Usage example
24
+ input_file = 't5xxl_fp16.safetensors' # Replace with your input model file
25
+ output_file = 't5xxl_fp16.safetensors' # Desired output file name
26
  reduce_key_size(input_file, output_file)