Files changed (1) hide show
  1. src/app.py +4 -4
src/app.py CHANGED
@@ -17,13 +17,13 @@ def get_results(model_name: str, library: str, options: list, access_token: str)
17
  stages = {"model": [], "gradients": [], "optimizer": [], "step": []}
18
  for i, option in enumerate(data):
19
  for stage in stages:
20
- stages[stage].append(option["Training using Adam (Peek vRAM)"][stage])
21
- value = max(data[i]["Training using Adam (Peek vRAM)"].values())
22
  if value == -1:
23
  value = "N/A"
24
  else:
25
  value = convert_bytes(value)
26
- data[i]["Training using Adam (Peek vRAM)"] = value
27
 
28
  if any(value != -1 for value in stages["model"]):
29
  out_explain = "## Training using Adam explained:\n"
@@ -73,7 +73,7 @@ with gr.Blocks() as demo:
73
  )
74
  out_text = gr.Markdown()
75
  out = gr.DataFrame(
76
- headers=["dtype", "Largest Layer", "Total Size", "Training using Adam (Peek vRAM)"],
77
  interactive=False,
78
  visible=False,
79
  )
 
17
  stages = {"model": [], "gradients": [], "optimizer": [], "step": []}
18
  for i, option in enumerate(data):
19
  for stage in stages:
20
+ stages[stage].append(option["Training using Adam (Peak vRAM)"][stage])
21
+ value = max(data[i]["Training using Adam (Peak vRAM)"].values())
22
  if value == -1:
23
  value = "N/A"
24
  else:
25
  value = convert_bytes(value)
26
+ data[i]["Training using Adam (Peak vRAM)"] = value
27
 
28
  if any(value != -1 for value in stages["model"]):
29
  out_explain = "## Training using Adam explained:\n"
 
73
  )
74
  out_text = gr.Markdown()
75
  out = gr.DataFrame(
76
+ headers=["dtype", "Largest Layer", "Total Size", "Training using Adam (Peak vRAM)"],
77
  interactive=False,
78
  visible=False,
79
  )