Files
vllm/cacheflow/models/sample.py