Skip to content

Commit fafbe11

Browse files
authored
[Docs] Fix griffe warnings in vllm/lora/ops (#25369)
Signed-off-by: windsonsea <[email protected]>
1 parent 78237e4 commit fafbe11

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

vllm/lora/ops/triton_ops/lora_kernel_metadata.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -83,8 +83,8 @@ def prepare_tensors(self, token_lora_mapping: torch.Tensor) -> None:
8383
Prepare kernel metadata tensors for the current forward pass.
8484
8585
Args:
86-
token_lora_tensor (torch.Tensor): Tensor containing lora indices
87-
for each input token.
86+
token_lora_mapping (torch.Tensor): Tensor containing lora indices
87+
for each input token.
8888
"""
8989

9090
self._reset()
@@ -136,7 +136,7 @@ def meta_args(
136136
137137
Args:
138138
token_nums (int): Number of input tokens in the current forward
139-
pass.
139+
pass of the kernel.
140140
"""
141141
return (
142142
self.token_lora_mapping[:token_nums],

vllm/lora/ops/xla_ops/lora_ops.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,6 @@ def bgmv_shrink(
9393
inputs (torch.Tensor): Input tensor of shape [num_tokens, hidden_size].
9494
lora_b_weights (torch.Tensor): LoRA weights of shape
9595
[num_loras, lora_rank, hidden_size].
96-
output_tensor (torch.Tensor): (Unused) output tensor (placeholder).
9796
lora_indices_tensor (torch.Tensor): Tensor of shape [num_tokens]
9897
indicating which LoRA matrix to use for each token.
9998
scaling (float, optional): Scalar multiplier applied to the output.

0 commit comments

Comments
 (0)