Omartificial-Intelligence-Space commited on
Commit
dd5dfbd
·
verified ·
1 Parent(s): 4ab3edc

update utils

Browse files
Files changed (1) hide show
  1. 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="model", # Ensure this matches the DataFrame's 'model' column
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
- "model", "average", "model_type", "weight_type", "precision", "license", "likes", "still_on_hub"
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