BenchmarkBot commited on
Commit
c5c2773
β€’
1 Parent(s): bc255c5

transform energy tokens to int

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -84,9 +84,9 @@ def get_benchmark_df(benchmark="Succeeded-1xA100-80GB"):
84
  clusters_df, left_on="model", right_on="best_scored_model"
85
  )
86
  # transpose energy consumption
87
- merged_df["generate.energy_consumption(tokens/kWh)"] = 1 / merged_df[
88
- "generate.energy_consumption(kWh/token)"
89
- ].fillna(1)
90
  # fix nan values
91
  merged_df.loc[
92
  merged_df["generate.energy_consumption(tokens/kWh)"] == 1,
 
84
  clusters_df, left_on="model", right_on="best_scored_model"
85
  )
86
  # transpose energy consumption
87
+ merged_df["generate.energy_consumption(tokens/kWh)"] = (
88
+ 1 / merged_df["generate.energy_consumption(kWh/token)"].fillna(1)
89
+ ).astype(int)
90
  # fix nan values
91
  merged_df.loc[
92
  merged_df["generate.energy_consumption(tokens/kWh)"] == 1,