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
76ed5340
Unverified
Commit
76ed5340
authored
Oct 28, 2024
by
youkaichao
Committed by
GitHub
Oct 28, 2024
Browse files
[torch.compile] add deepseek v2 compile (#9775)
Signed-off-by:
youkaichao
<
youkaichao@gmail.com
>
parent
97b61bfa
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
0 deletions
+2
-0
vllm/model_executor/models/deepseek_v2.py
vllm/model_executor/models/deepseek_v2.py
+2
-0
No files found.
vllm/model_executor/models/deepseek_v2.py
View file @
76ed5340
...
...
@@ -28,6 +28,7 @@ from torch import nn
from
transformers
import
PretrainedConfig
from
vllm.attention
import
Attention
,
AttentionMetadata
from
vllm.compilation.decorators
import
support_torch_compile
from
vllm.config
import
CacheConfig
from
vllm.distributed
import
(
get_pp_group
,
get_tensor_model_parallel_world_size
,
...
...
@@ -403,6 +404,7 @@ class DeepseekV2DecoderLayer(nn.Module):
return
hidden_states
,
residual
@
support_torch_compile
class
DeepseekV2Model
(
nn
.
Module
):
fall_back_to_pt_during_load
=
False
...
...
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