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
bfab2196
Unverified
Commit
bfab2196
authored
Aug 28, 2025
by
Jiangyun Zhu
Committed by
GitHub
Aug 28, 2025
Browse files
[Model] [gpt-oss] fix gpt-oss pp support (#23815)
Signed-off-by:
zjy0516
<
riverclouds.zhu@qq.com
>
parent
a3432f18
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
3 deletions
+2
-3
vllm/model_executor/models/gpt_oss.py
vllm/model_executor/models/gpt_oss.py
+2
-3
No files found.
vllm/model_executor/models/gpt_oss.py
View file @
bfab2196
...
...
@@ -668,9 +668,8 @@ class GptOssForCausalLM(nn.Module, SupportsPP):
positions
:
torch
.
Tensor
,
intermediate_tensors
:
Optional
[
IntermediateTensors
]
=
None
,
inputs_embeds
:
Optional
[
torch
.
Tensor
]
=
None
)
->
torch
.
Tensor
:
assert
intermediate_tensors
is
None
assert
inputs_embeds
is
None
return
self
.
model
(
input_ids
,
positions
)
return
self
.
model
(
input_ids
,
positions
,
intermediate_tensors
,
inputs_embeds
)
def
compute_logits
(
self
,
hidden_states
:
torch
.
Tensor
,
sampling_metadata
:
SamplingMetadata
)
->
torch
.
Tensor
:
...
...
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