Bank
35 / 87
InferenceShort answerHard

You serve a 70B model with a 1B draft model. The draft proposes tokens one at a time, each draft step costing about 5% of a 70B forward pass, and the 70B model then scores all in one forward pass that costs about as much as generating one token, since decoding is memory-bound.

Left to right, each draft token is kept with probability , where and are the 70B and draft models' probabilities for it. The first rejected token is resampled from the 70B model and the rest of the draft is discarded; if all survive, the 70B pass adds one more token.

How would you pick the draft length for this pair of models, and what limits the speedup as grows?