updating link

vllm-project · Jan 5, 2024 · 434c525 · 434c525
1 parent 876bf2e
commit 434c525
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/vllm/model_executor/layers/rejection_sampler.py b/vllm/model_executor/layers/rejection_sampler.py
@@ -214,8 +214,8 @@ def _get_recovered_probs(
         .. math::
             (f(x))_+ = \frac{\max(0, f(x))}{\sum_x \max(0, f(x))}
 
-        See https://github.com/anyscale/vllm/pull/157 for a visualization of the
-        draft, target, and recovered probability distributions.
+        See https://github.com/vllm-project/vllm/pull/2336 for a visualization
+        of the draft, target, and recovered probability distributions.
 
         Returns a tensor of shape [batch_size, k, vocab_size].