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
5f71c3f8
Commit
5f71c3f8
authored
Jun 21, 2025
by
zhuwenwen
Browse files
support ndhwc of qwen2-vl and qwen2.5-vl
parent
7996b363
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
2 additions
and
0 deletions
+2
-0
vllm/model_executor/models/qwen2_5_vl.py
vllm/model_executor/models/qwen2_5_vl.py
+1
-0
vllm/model_executor/models/qwen2_vl.py
vllm/model_executor/models/qwen2_vl.py
+1
-0
No files found.
vllm/model_executor/models/qwen2_5_vl.py
View file @
5f71c3f8
...
@@ -436,6 +436,7 @@ class Qwen2_5_VisionPatchEmbed(nn.Module):
...
@@ -436,6 +436,7 @@ class Qwen2_5_VisionPatchEmbed(nn.Module):
L
,
C
=
x
.
shape
L
,
C
=
x
.
shape
x
=
x
.
view
(
L
,
-
1
,
self
.
temporal_patch_size
,
self
.
patch_size
,
x
=
x
.
view
(
L
,
-
1
,
self
.
temporal_patch_size
,
self
.
patch_size
,
self
.
patch_size
)
self
.
patch_size
)
x
=
x
.
to
(
memory_format
=
torch
.
channels_last_3d
)
x
=
self
.
proj
(
x
).
view
(
L
,
self
.
hidden_size
)
x
=
self
.
proj
(
x
).
view
(
L
,
self
.
hidden_size
)
return
x
return
x
...
...
vllm/model_executor/models/qwen2_vl.py
View file @
5f71c3f8
...
@@ -461,6 +461,7 @@ class Qwen2VisionPatchEmbed(nn.Module):
...
@@ -461,6 +461,7 @@ class Qwen2VisionPatchEmbed(nn.Module):
L
,
C
=
x
.
shape
L
,
C
=
x
.
shape
x
=
x
.
view
(
L
,
-
1
,
self
.
temporal_patch_size
,
self
.
patch_size
,
x
=
x
.
view
(
L
,
-
1
,
self
.
temporal_patch_size
,
self
.
patch_size
,
self
.
patch_size
)
self
.
patch_size
)
# x=x.to(memory_format=torch.channels_last_3d)
x
=
self
.
proj
(
x
).
view
(
L
,
self
.
embed_dim
)
x
=
self
.
proj
(
x
).
view
(
L
,
self
.
embed_dim
)
return
x
return
x
...
...
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