Fixing compute logits in vllm adapter.#2907

copybara-service[bot] merged 1 commit intomainAI-Hypercomputer/maxtext:mainfrom

nicogrande/vllm-adapter-compute-logits-fixAI-Hypercomputer/maxtext:nicogrande/vllm-adapter-compute-logits-fixCopy head branch name to clipboard