New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

typo: fix target_probs docs after uniform_samples removal #935

Merged

yzh119 merged 1 commit into flashinfer-ai:main from markmc:typo-target-probs-docs

Mar 12, 2025

Contributor

markmc commented Mar 11, 2025

Looks like a typo in #912

-    uniform_samples: torch.Tensor
-        The uniform samples used as needle for sampling, shape ``(batch_size, num_speculate_tokens + 1)``.
+    target_probs: torch.Tensor
         Expected to be uniformly distributed in ``[0, 1)``.
      target_probs: torch.Tensor
         The probability over vocabulary generated by target model.


          typo: fix target_probs docs after uniform_samples removal

19a1761

yzh119 approved these changes

View reviewed changes

Collaborator

yzh119 left a comment

Thanks for the fix, these text belong to legacy APIs which is no longer required, thank you!

yzh119 merged commit d462a9d into flashinfer-ai:main

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet