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
|