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
064cac7b
Unverified
Commit
064cac7b
authored
Sep 18, 2025
by
Nikhil Gupta
Committed by
GitHub
Sep 18, 2025
Browse files
[fix]: remove data type hardcoding from gptoss model implementation (#23807)
Signed-off-by:
Nikhil Gupta
<
nikhil.gupta2@arm.com
>
parent
e19bce40
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
3 deletions
+1
-3
vllm/model_executor/models/gpt_oss.py
vllm/model_executor/models/gpt_oss.py
+1
-3
No files found.
vllm/model_executor/models/gpt_oss.py
View file @
064cac7b
...
...
@@ -76,7 +76,6 @@ class OAIAttention(nn.Module):
self
.
sinks
=
torch
.
nn
.
Parameter
(
torch
.
empty
(
config
.
num_attention_heads
//
tp_size
,
dtype
=
torch
.
bfloat16
,
requires_grad
=
False
))
self
.
q_size
=
self
.
num_attention_heads
*
self
.
head_dim
//
tp_size
...
...
@@ -145,8 +144,7 @@ class MLPBlock(torch.nn.Module):
self
.
experts_per_token
=
config
.
num_experts_per_tok
self
.
world_size
=
dist
.
get_world_size
()
if
dist
.
is_initialized
()
else
1
self
.
router
=
torch
.
nn
.
Linear
(
config
.
hidden_size
,
config
.
num_local_experts
,
dtype
=
torch
.
bfloat16
)
config
.
num_local_experts
)
assert
config
.
intermediate_size
%
self
.
world_size
==
0
self
.
experts
=
FusedMoE
(
num_experts
=
config
.
num_local_experts
,
top_k
=
config
.
num_experts_per_tok
,
...
...
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