update utils
Browse files- src/display/utils.py +3 -3
src/display/utils.py
CHANGED
|
@@ -22,7 +22,7 @@ COLUMNS: List[ColumnContent] = []
|
|
| 22 |
# Essential columns
|
| 23 |
COLUMNS.append(
|
| 24 |
ColumnContent(
|
| 25 |
-
name="
|
| 26 |
type=str,
|
| 27 |
label="Model",
|
| 28 |
description="Model name",
|
|
@@ -99,7 +99,7 @@ COLUMNS.extend([
|
|
| 99 |
# Create lists of column names for use in the application
|
| 100 |
COLS = [col.name for col in COLUMNS]
|
| 101 |
BENCHMARK_COLS = [col.name for col in COLUMNS if col.name not in [
|
| 102 |
-
"
|
| 103 |
]]
|
| 104 |
|
| 105 |
# For the queue columns in the submission tab
|
|
@@ -199,4 +199,4 @@ class Precision(Enum):
|
|
| 199 |
return Precision.float16
|
| 200 |
if precision_str in ["torch.bfloat16", "bfloat16"]:
|
| 201 |
return Precision.bfloat16
|
| 202 |
-
return Precision.Unknown
|
|
|
|
| 22 |
# Essential columns
|
| 23 |
COLUMNS.append(
|
| 24 |
ColumnContent(
|
| 25 |
+
name="model_name", # Changed from "model" to "model_name"
|
| 26 |
type=str,
|
| 27 |
label="Model",
|
| 28 |
description="Model name",
|
|
|
|
| 99 |
# Create lists of column names for use in the application
|
| 100 |
COLS = [col.name for col in COLUMNS]
|
| 101 |
BENCHMARK_COLS = [col.name for col in COLUMNS if col.name not in [
|
| 102 |
+
"model_name", "average", "model_type", "weight_type", "precision", "license", "likes", "still_on_hub"
|
| 103 |
]]
|
| 104 |
|
| 105 |
# For the queue columns in the submission tab
|
|
|
|
| 199 |
return Precision.float16
|
| 200 |
if precision_str in ["torch.bfloat16", "bfloat16"]:
|
| 201 |
return Precision.bfloat16
|
| 202 |
+
return Precision.Unknown
|