From 56e60ad5a6692d4cee38701237de1bb37e19e994 Mon Sep 17 00:00:00 2001 From: Awni Hannun Date: Wed, 26 Feb 2025 15:44:57 -0800 Subject: [PATCH] fix manage for new transformers (#1304) --- llms/mlx_lm/manage.py | 17 ++++++++++++++++- 1 file changed, 16 insertions(+), 1 deletion(-) diff --git a/llms/mlx_lm/manage.py b/llms/mlx_lm/manage.py index 9827f3dc..c06de6b3 100644 --- a/llms/mlx_lm/manage.py +++ b/llms/mlx_lm/manage.py @@ -2,7 +2,22 @@ import argparse from typing import List, Union from huggingface_hub import scan_cache_dir -from transformers.commands.user import tabulate + + +def tabulate(rows: List[List[Union[str, int]]], headers: List[str]) -> str: + """ + Inspired by: + - stackoverflow.com/a/8356620/593036 + - stackoverflow.com/questions/9535954/printing-lists-as-tabular-data + """ + col_widths = [max(len(str(x)) for x in col) for col in zip(*rows, headers)] + row_format = ("{{:{}}} " * len(headers)).format(*col_widths) + lines = [] + lines.append(row_format.format(*headers)) + lines.append(row_format.format(*["-" * w for w in col_widths])) + for row in rows: + lines.append(row_format.format(*row)) + return "\n".join(lines) def ask_for_confirmation(message: str) -> bool: