Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
fafbe11a
Unverified
Commit
fafbe11a
authored
Sep 23, 2025
by
Michael Yao
Committed by
GitHub
Sep 23, 2025
Browse files
[Docs] Fix griffe warnings in vllm/lora/ops (#25369)
Signed-off-by:
windsonsea
<
haifeng.yao@daocloud.io
>
parent
78237e43
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
3 additions
and
4 deletions
+3
-4
vllm/lora/ops/triton_ops/lora_kernel_metadata.py
vllm/lora/ops/triton_ops/lora_kernel_metadata.py
+3
-3
vllm/lora/ops/xla_ops/lora_ops.py
vllm/lora/ops/xla_ops/lora_ops.py
+0
-1
No files found.
vllm/lora/ops/triton_ops/lora_kernel_metadata.py
View file @
fafbe11a
...
@@ -83,8 +83,8 @@ class LoRAKernelMeta:
...
@@ -83,8 +83,8 @@ class LoRAKernelMeta:
Prepare kernel metadata tensors for the current forward pass.
Prepare kernel metadata tensors for the current forward pass.
Args:
Args:
token_lora_
tensor
(torch.Tensor): Tensor containing lora indices
token_lora_
mapping
(torch.Tensor): Tensor containing lora indices
for each input token.
for each input token.
"""
"""
self
.
_reset
()
self
.
_reset
()
...
@@ -136,7 +136,7 @@ class LoRAKernelMeta:
...
@@ -136,7 +136,7 @@ class LoRAKernelMeta:
Args:
Args:
token_nums (int): Number of input tokens in the current forward
token_nums (int): Number of input tokens in the current forward
pass.
pass of the kernel.
"""
"""
return
(
return
(
self
.
token_lora_mapping
[:
token_nums
],
self
.
token_lora_mapping
[:
token_nums
],
...
...
vllm/lora/ops/xla_ops/lora_ops.py
View file @
fafbe11a
...
@@ -93,7 +93,6 @@ def bgmv_shrink(
...
@@ -93,7 +93,6 @@ def bgmv_shrink(
inputs (torch.Tensor): Input tensor of shape [num_tokens, hidden_size].
inputs (torch.Tensor): Input tensor of shape [num_tokens, hidden_size].
lora_b_weights (torch.Tensor): LoRA weights of shape
lora_b_weights (torch.Tensor): LoRA weights of shape
[num_loras, lora_rank, hidden_size].
[num_loras, lora_rank, hidden_size].
output_tensor (torch.Tensor): (Unused) output tensor (placeholder).
lora_indices_tensor (torch.Tensor): Tensor of shape [num_tokens]
lora_indices_tensor (torch.Tensor): Tensor of shape [num_tokens]
indicating which LoRA matrix to use for each token.
indicating which LoRA matrix to use for each token.
scaling (float, optional): Scalar multiplier applied to the output.
scaling (float, optional): Scalar multiplier applied to the output.
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment