Top-K Sampling

Sampling from only the K most likely next tokens.