Update app.py

app.py

@@ -62,7 +62,7 @@ c2.write(str(mqa_time))
 
 st.header('Attention')
 mha_flop = 2*bs*h*(d/h)*n
-mha_bytes = 2*bs*h*(d/h) + 2*bs*h*n*(d/h)
+mha_bytes = 2*bs*h*(d/h) + 2*bs*h*n*(d/h) + 2*bs*h*n
 mha_int = mha_flop/mha_bytes
 mha_time = (mha_flop/TFLOPS + mha_bytes/GB_S)*1000
 
@@ -82,7 +82,7 @@ c1.write("Time (ms):")
 c2.write(str(mha_time))
 
 mqa_flop = 2*bs*h*(d/h)*n
-mqa_bytes = 2*bs*h*(d/h) + 2*bs*n*(d/h)
+mqa_bytes = 2*bs*h*(d/h) + 2*bs*n*(d/h) + 2*bs*h*n
 mqa_intensity = mqa_flop/mqa_bytes
 mqa_time = (mqa_flop/TFLOPS + mqa_bytes/GB_S)*1000