Files
vllm/vllm/v1/sample/rejection_sampler.py