Skip to content

Commit

Permalink
[Doc] Fix typo (#11666)
Browse files Browse the repository at this point in the history
Signed-off-by: Kazuhiro Serizawa <[email protected]>
  • Loading branch information
serihiro authored Jan 1, 2025
1 parent f962f42 commit 6d70198
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion vllm/model_executor/layers/rejection_sampler.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@ def __init__(self,
strict_mode: Whether or not to perform shape/device/dtype checks
during sampling. This catches correctness issues but adds
nontrivial latency.
use_falshinfer: We will use this parameter to determine whether
use_flashinfer: We will use this parameter to determine whether
to use the FlashInfer rejection sampling kernel or not. If it's
None, we will use the default value from the environment variable.
This parameter is only used for testing purposes.
Expand Down
2 changes: 1 addition & 1 deletion vllm/v1/sample/ops/topk_topp_sampler.py
Original file line number Diff line number Diff line change
Expand Up @@ -44,7 +44,7 @@ def __init__(self):
logger.warning(
"FlashInfer is not available. Falling back to the PyTorch-"
"native implementation of top-p & top-k sampling. For the "
"best performance, please install FalshInfer.")
"best performance, please install FlashInfer.")
self.forward = self.forward_native
else:
self.forward = self.forward_native
Expand Down

0 comments on commit 6d70198

Please sign in to comment.