Update app.py
Browse files
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 |
-
|
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
|