Update rp.py
Browse files
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.
|
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 = '
|
25 |
-
output_file = '
|
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)
|