Update app.py

app.py

@@ -120,7 +120,8 @@ st.write("Multi-Query Attention: " + str(mqa_total_time))
 st.write("Speed-up MQA over MHA: " + str(mha_total_time/mqa_total_time))
 
 st.header("Memory consumption")
-
+num_params = 12*l*d*d
+st.write("Num Parameters (in B)" + str(round(num_params/1e9, 3)))
 
 
 breakdown = st.checkbox("Show breakdown per operation")