File tree 1 file changed +2
-2
lines changed
tools/llm_weight_compression
1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -167,7 +167,7 @@ def visualize_experiments(model_id: str, params_grid: List[Params]):
167
167
:param params_grid:
168
168
"""
169
169
rows = [[model_id , params .get_key ()] for params in params_grid ]
170
- print ("List of configurations to test out:" )
170
+ print (f "List of configurations to test out ( { len ( params_grid ) } ) :" )
171
171
print (tabulate (tabular_data = rows , headers = ["Model ID" , "Experiment" ], tablefmt = "mixed_grid" ))
172
172
173
173
@@ -577,7 +577,7 @@ def main():
577
577
578
578
# --------- Compress ---------
579
579
compression_config = config ["compression" ]
580
- compress (args .model_id , ROOT_MODEL_DIR , compression_config )
580
+ compress (args .model_id , ROOT_MODEL_DIR , compression_config , show_only = args . show_only )
581
581
582
582
if args .show_only :
583
583
return
You can’t perform that action at this time.
0 commit comments