added number of params

app.py

@@ -5,7 +5,7 @@ import plotly.express as px
 from apscheduler.schedulers.background import BackgroundScheduler
 
 from src.assets.text_content import TITLE, INTRODUCTION_TEXT, SINGLE_A100_TEXT, CITATION_BUTTON_LABEL, CITATION_BUTTON_TEXT
-from src.utils import restart_space, load_dataset_repo, make_clickable_model, make_clickable_score
+from src.utils import restart_space, load_dataset_repo, make_clickable_model, make_clickable_score, num_to_str
 from src.assets.css_html_js import custom_css
 
 
@@ -17,12 +17,13 @@ COLUMNS_MAPPING = {
     "model": "Model 🤗",
     "backend.name": "Backend 🏭",
     "backend.torch_dtype": "Load Dtype 📥",
+    "num_parameters": "#Parameters 📏",
     "forward.peak_memory(MB)": "Peak Memory (MB) ⬇️",
     "generate.throughput(tokens/s)": "Throughput (tokens/s) ⬆️",
     "h4_score": "Average Open LLM Score ⬆️",
-
 }
-COLUMNS_DATATYPES = ["markdown", "str", "str", "number", "number", "markdown"]
+COLUMNS_DATATYPES = ["markdown", "str", "str",
+                     "number", "number", "number", "markdown"]
 SORTING_COLUMN = ["Throughput (tokens/s) ⬆️"]
 
 
@@ -55,6 +56,7 @@ def get_benchmark_table(bench_df):
     bench_df["Model 🤗"] = bench_df["Model 🤗"].apply(make_clickable_model)
     bench_df["Average Open LLM Score ⬆️"] = bench_df["Average Open LLM Score ⬆️"].apply(
         make_clickable_score)
+    bench_df["#Parameters 📏"] = bench_df["#Parameters 📏"].apply(num_to_str)
 
     return bench_df
 

src/utils.py

@@ -66,3 +66,15 @@ def make_clickable_model(model_name):
 def make_clickable_score(score):
     link = f"https://huggingface.co/spaces/HuggingFaceH4/open_llm_leaderboard"
     return f'<a target="_blank" href="{link}" style="color: var(--link-text-color); text-decoration: underline;text-decoration-style: dotted;">{score}</a>'
+
+
+def num_to_str(num):
+    if num < 1000:
+        return str(int(num))
+    elif num < 1000000:
+        return str(int(num / 1000)) + "K"
+    elif num < 1000000000:
+        return str(int(num / 1000000)) + "M"
+    elif num < 1000000000000:
+        return str(int(num / 1000000000)) + "B"
+    return None