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
87bc0c49
Unverified
Commit
87bc0c49
authored
Oct 17, 2025
by
Jee Jee Li
Committed by
GitHub
Oct 17, 2025
Browse files
[Bugfix] Fix ReplicatedLinearWithLoRA (#27065)
Signed-off-by:
Jee Jee Li
<
pandaleefree@gmail.com
>
parent
fe3b9372
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
12 additions
and
0 deletions
+12
-0
vllm/lora/layers/replicated_linear.py
vllm/lora/layers/replicated_linear.py
+12
-0
No files found.
vllm/lora/layers/replicated_linear.py
View file @
87bc0c49
...
...
@@ -56,3 +56,15 @@ class ReplicatedLinearWithLoRA(BaseLinearLayerWithLoRA):
model_config
:
PretrainedConfig
|
None
,
)
->
bool
:
return
type
(
source_layer
)
is
ReplicatedLinear
def
slice_lora_a
(
self
,
lora_a
:
torch
.
Tensor
|
list
[
torch
.
Tensor
|
None
]
)
->
torch
.
Tensor
|
list
[
torch
.
Tensor
|
None
]:
"""Slice lora a if splitting for tensor parallelism."""
return
lora_a
def
slice_lora_b
(
self
,
lora_b
:
torch
.
Tensor
|
list
[
torch
.
Tensor
|
None
]
)
->
torch
.
Tensor
|
list
[
torch
.
Tensor
|
None
]:
"""Slice lora b if splitting with tensor parallelism."""
return
lora_b
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