2 ms·

If you want to be able to generate up to 128k tokens in one go successfully, then yes, that math checks out.