Skip to content

Commit

Permalink
minor improvements
Browse files Browse the repository at this point in the history
  • Loading branch information
andrey-churkin committed Feb 20, 2025
1 parent b5f72d5 commit 3e4cfca
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tools/llm_weight_compression/run.py
Original file line number Diff line number Diff line change
Expand Up @@ -167,7 +167,7 @@ def visualize_experiments(model_id: str, params_grid: List[Params]):
:param params_grid:
"""
rows = [[model_id, params.get_key()] for params in params_grid]
print("List of configurations to test out:")
print(f"List of configurations to test out ({len(params_grid)}):")
print(tabulate(tabular_data=rows, headers=["Model ID", "Experiment"], tablefmt="mixed_grid"))


Expand Down Expand Up @@ -577,7 +577,7 @@ def main():

# --------- Compress ---------
compression_config = config["compression"]
compress(args.model_id, ROOT_MODEL_DIR, compression_config)
compress(args.model_id, ROOT_MODEL_DIR, compression_config, show_only=args.show_only)

if args.show_only:
return
Expand Down

0 comments on commit 3e4cfca

Please sign in to comment.