Jellyfish042 commited on
Commit
ddb0a48
·
1 Parent(s): 4c16ce5

Mistral-Nemo-Base-2407

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -194,6 +194,7 @@ def update_table(period: str,
194
  col not in ['Name', 'Parameters Count (B)', 'Average (lower=better)']])
195
 
196
  styler = filtered_data.style.format(formatter)
 
197
  for column in target_color_columns:
198
  if column in vmin and column in vmax: # Ensure that the vmin and vmax dicts contain the column
199
  styler = styler.background_gradient(cmap=cmap, subset=[column], vmin=vmin[column], vmax=vmax[column])
@@ -271,7 +272,8 @@ def create_scaling_plot(all_data, period):
271
  names_to_connect = ['Meta-Llama-3-8B',
272
  'stablelm-3b-4e1t',
273
  'Qwen2-1.5B',
274
- 'TinyLlama-1.1B-intermediate-step-1431k-3T']
 
275
  connection_points = new_df[new_df['Name'].isin(names_to_connect)]
276
 
277
  new_df['Color'] = new_df['Name'].apply(lambda name: '#39C5BB' if name in names_to_connect else '#636efa')
 
194
  col not in ['Name', 'Parameters Count (B)', 'Average (lower=better)']])
195
 
196
  styler = filtered_data.style.format(formatter)
197
+
198
  for column in target_color_columns:
199
  if column in vmin and column in vmax: # Ensure that the vmin and vmax dicts contain the column
200
  styler = styler.background_gradient(cmap=cmap, subset=[column], vmin=vmin[column], vmax=vmax[column])
 
272
  names_to_connect = ['Meta-Llama-3-8B',
273
  'stablelm-3b-4e1t',
274
  'Qwen2-1.5B',
275
+ 'TinyLlama-1.1B-intermediate-step-1431k-3T',
276
+ 'Mistral-Nemo-Base-2407']
277
  connection_points = new_df[new_df['Name'].isin(names_to_connect)]
278
 
279
  new_df['Color'] = new_df['Name'].apply(lambda name: '#39C5BB' if name in names_to_connect else '#636efa')