prefill bottlenecks on longer prompts

[batchsize]

I'd start with the simpler stack.

My first guess would be this: Before rebuilding anything, I'd test one clean baseline with Mistral and compare from there. Once you know the plain path works, then add the faster path back in piece by piece.