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
4497c8f8
Unverified
Commit
4497c8f8
authored
Oct 13, 2025
by
XiongfeiWei
Committed by
GitHub
Oct 14, 2025
Browse files
Fix lora tests failure in TPU CI due to the removal of LoRA bias (#26723)
Signed-off-by:
Xiongfei Wei
<
isaacwxf23@gmail.com
>
parent
2e36cdbe
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
2 deletions
+1
-2
vllm/v1/worker/tpu_model_runner.py
vllm/v1/worker/tpu_model_runner.py
+1
-2
No files found.
vllm/v1/worker/tpu_model_runner.py
View file @
4497c8f8
...
...
@@ -2128,12 +2128,11 @@ def replace_set_lora(model):
lora_a
:
torch
.
Tensor
,
lora_b
:
torch
.
Tensor
,
embeddings_tensor
:
torch
.
Tensor
|
None
,
bias
:
torch
.
Tensor
|
None
=
None
,
):
# TODO: The integer index leads to a recompilation, but converting it
# to a tensor doesn't seem to work anymore. This might be fixed with a
# later release of torch_xla.
self
.
_original_set_lora
(
index
,
lora_a
,
lora_b
,
embeddings_tensor
,
bias
)
self
.
_original_set_lora
(
index
,
lora_a
,
lora_b
,
embeddings_tensor
)
torch_xla
.
sync
(
wait
=
False
)
def
_tpu_reset_lora
(
self
,
index
:
int
):
...
...
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