Update app.py

app.py

@@ -344,7 +344,8 @@ def Compute_ALL_Model_memory(vocab_size, layer_num, hidden_size, ffn_size, seq_l
     numParameters = round(Get_BillionParameter(numParameters), 3)
     model_states_memory = round(Get_GigaByte(model_states_memory), 3)
     activation_memory = round(Get_GigaByte(activation_memory), 3)
-    Total = round(model_states_memory + activation_memory, 3)
+    other_memory = 5
+    Total = round(model_states_memory + activation_memory + other_memory, 3)
 
     # record
     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, 
@@ -363,6 +364,7 @@ def Compute_ALL_Model_memory(vocab_size, layer_num, hidden_size, ffn_size, seq_l
                 Model parameters on each device = {str(numParameters)} B, \n
                 Model_states = Weight + Gradient + Optimizer = {str(model_states_memory)} GB, \n
                 Activation = {str(activation_memory)} GB, \n
+                Other memory = 5 GB, \n
                 Total memory consumption = {str(Total)} GB \n
            """, record_df, count