round params

app.py

@@ -76,7 +76,7 @@ def init_leaderboard(dataframe):
             # ColumnFilter(AutoEvalColumn.model_type.name, type="checkboxgroup", label="Model types"),
             # ColumnFilter(AutoEvalColumn.precision.name, type="checkboxgroup", label="Precision"),
             ColumnFilter(
-                AutoEvalColumn.params.name,
+                AutoEvalColumn.round_params.name,
                 type="slider",
                 min=1,
                 max=9,

src/display/utils.py

@@ -37,6 +37,7 @@ for task in Tasks:
 # auto_eval_column_dict.append(["precision", ColumnContent, ColumnContent("Precision", "str", False)])
 # auto_eval_column_dict.append(["license", ColumnContent, ColumnContent("Hub License", "str", False)])
 auto_eval_column_dict.append(["params", ColumnContent, ColumnContent("#Params (B)", "number", False)])
+auto_eval_column_dict.append(["round_params", ColumnContent, ColumnContent("round_params", "number", False)])
 # auto_eval_column_dict.append(["likes", ColumnContent, ColumnContent("Hub ❤️", "number", False)])
 # auto_eval_column_dict.append(["still_on_hub", ColumnContent, ColumnContent("Available on the hub", "bool", False)])
 # auto_eval_column_dict.append(["revision", ColumnContent, ColumnContent("Model sha", "str", False, False)])

src/leaderboard/read_evals.py

@@ -115,9 +115,11 @@ class EvalResult:
         """Converts the Eval Result to a dict compatible with our dataframe display"""
         average = sum([v for v in self.results.values() if v is not None]) / len(Tasks)
 
-        if self.num_params % 0.5 == 0:
-            self.num_params -= 0.01
-        self.num_params = round(self.num_params)
+        # if self.num_params % 0.5 == 0:
+        #     self.num_params -= 0.01
+        # self.num_params = round(self.num_params)
+
+        self.round_params = math.floor(self.num_params)
 
         data_dict = {
             "eval_name": self.eval_name,  # not a column, just a save name,
@@ -132,6 +134,7 @@ class EvalResult:
             # AutoEvalColumn.license.name: self.license,
             # AutoEvalColumn.likes.name: self.likes,
             AutoEvalColumn.params.name: self.num_params,
+            AutoEvalColumn.round_params.name: self.round_params,
             # AutoEvalColumn.still_on_hub.name: self.still_on_hub,
         }