Skip to content

Commit b447f7e

Browse files
authored
Fix qwen vl (#3096)
* Fixing qwen2.5 VL. * Fixing the CI.
1 parent 094975c commit b447f7e

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

Cargo.lock

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

server/text_generation_server/models/custom_modeling/qwen2_5_vl.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -313,7 +313,6 @@ def __init__(
313313

314314

315315
class Qwen2_5_VLConfig(PretrainedConfig):
316-
317316
def __init__(
318317
self,
319318
vocab_size=152064,
@@ -702,7 +701,6 @@ def forward(
702701
pixel_values: torch.Tensor,
703702
grid_thw: Optional[torch.LongTensor] = None,
704703
) -> torch.Tensor:
705-
706704
# reshape the input tensor for processing
707705
shape = (
708706
-1,
@@ -967,6 +965,7 @@ def forward(
967965
max_s=max_s,
968966
true_max_s=max_s,
969967
prefill_cache_indices=prefill_cache_indices,
968+
adapter_data=adapter_data,
970969
)
971970
if lm_head_indices is not None:
972971
hidden_states = hidden_states[lm_head_indices]

0 commit comments

Comments
 (0)