Skip to content

Commit 3e4cfca

Browse files
minor improvements
1 parent b5f72d5 commit 3e4cfca

File tree

1 file changed

+2
-2
lines changed
  • tools/llm_weight_compression

1 file changed

+2
-2
lines changed

tools/llm_weight_compression/run.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -167,7 +167,7 @@ def visualize_experiments(model_id: str, params_grid: List[Params]):
167167
:param params_grid:
168168
"""
169169
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)}):")
171171
print(tabulate(tabular_data=rows, headers=["Model ID", "Experiment"], tablefmt="mixed_grid"))
172172

173173

@@ -577,7 +577,7 @@ def main():
577577

578578
# --------- Compress ---------
579579
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)
581581

582582
if args.show_only:
583583
return

0 commit comments

Comments
 (0)