xxyux commited on
Commit
affd6ff
·
verified ·
1 Parent(s): aed6a9f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -340,7 +340,7 @@ def Compute_ALL_Model_memory(vocab_size, layer_num, hidden_size, ffn_size, seq_l
340
 
341
  # record
342
  new_row = pd.DataFrame([[layer_num, hidden_size, ffn_size, seq_length, head_num, group_query_num, dp, tp, pp, cp, gpu_num, b, is_fp8,
343
- numParameters, model_states_memory, activation_memory, Total]],
344
  columns=col)
345
  if count == 1:
346
  record_df = new_row
 
340
 
341
  # record
342
  new_row = pd.DataFrame([[layer_num, hidden_size, ffn_size, seq_length, head_num, group_query_num, dp, tp, pp, cp, gpu_num, b, is_fp8,
343
+ numParametersTotal, model_states_memory, activation_memory, Total]],
344
  columns=col)
345
  if count == 1:
346
  record_df = new_row