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
90fc8f66
Commit
90fc8f66
authored
Nov 19, 2025
by
zhuwenwen
Browse files
update VLLM_USE_PD_SPLIT=0
parent
d95a8fff
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
5 additions
and
3 deletions
+5
-3
vllm/envs.py
vllm/envs.py
+1
-1
vllm/model_executor/model_loader/utils.py
vllm/model_executor/model_loader/utils.py
+4
-2
No files found.
vllm/envs.py
View file @
90fc8f66
...
@@ -1157,7 +1157,7 @@ environment_variables: dict[str, Callable[[], Any]] = {
...
@@ -1157,7 +1157,7 @@ environment_variables: dict[str, Callable[[], Any]] = {
(
"true"
,
"1"
)),
(
"true"
,
"1"
)),
# vLLM will split prefill and decode, not mix up
# vLLM will split prefill and decode, not mix up
"VLLM_USE_PD_SPLIT"
:
"VLLM_USE_PD_SPLIT"
:
lambda
:
(
os
.
environ
.
get
(
"VLLM_USE_PD_SPLIT"
,
"
Tru
e"
).
lower
()
in
lambda
:
(
os
.
environ
.
get
(
"VLLM_USE_PD_SPLIT"
,
"
Fals
e"
).
lower
()
in
(
"true"
,
"1"
)),
(
"true"
,
"1"
)),
# vLLM will sync to avoid pp vmfault
# vLLM will sync to avoid pp vmfault
"VLLM_USE_PP_SYNC"
:
"VLLM_USE_PP_SYNC"
:
...
...
vllm/model_executor/model_loader/utils.py
View file @
90fc8f66
...
@@ -253,8 +253,9 @@ def get_model_architecture(
...
@@ -253,8 +253,9 @@ def get_model_architecture(
os
.
environ
[
'VLLM_USE_OPT_CAT'
]
=
'1'
os
.
environ
[
'VLLM_USE_OPT_CAT'
]
=
'1'
# if not envs.is_set("VLLM_USE_LIGHTOP_FILL_MOE_ALIGN"):
# if not envs.is_set("VLLM_USE_LIGHTOP_FILL_MOE_ALIGN"):
# os.environ['VLLM_USE_LIGHTOP_FILL_MOE_ALIGN'] = '1'
# os.environ['VLLM_USE_LIGHTOP_FILL_MOE_ALIGN'] = '1'
else
:
if
not
envs
.
is_set
(
"VLLM_USE_PD_SPLIT"
):
if
not
envs
.
is_set
(
"VLLM_USE_PD_SPLIT"
):
os
.
environ
[
'VLLM_USE_PD_SPLIT'
]
=
'
0
'
os
.
environ
[
'VLLM_USE_PD_SPLIT'
]
=
'
1
'
if
os
.
getenv
(
'GEMM_PAD'
)
!=
'1'
:
if
os
.
getenv
(
'GEMM_PAD'
)
!=
'1'
:
...
@@ -271,8 +272,9 @@ def get_model_architecture(
...
@@ -271,8 +272,9 @@ def get_model_architecture(
os
.
environ
[
'VLLM_USE_OPT_CAT'
]
=
'1'
os
.
environ
[
'VLLM_USE_OPT_CAT'
]
=
'1'
# if not envs.is_set("VLLM_USE_LIGHTOP_FILL_MOE_ALIGN"):
# if not envs.is_set("VLLM_USE_LIGHTOP_FILL_MOE_ALIGN"):
# os.environ['VLLM_USE_LIGHTOP_FILL_MOE_ALIGN'] = '1'
# os.environ['VLLM_USE_LIGHTOP_FILL_MOE_ALIGN'] = '1'
else
:
if
not
envs
.
is_set
(
"VLLM_USE_PD_SPLIT"
):
if
not
envs
.
is_set
(
"VLLM_USE_PD_SPLIT"
):
os
.
environ
[
'VLLM_USE_PD_SPLIT'
]
=
'
0
'
os
.
environ
[
'VLLM_USE_PD_SPLIT'
]
=
'
1
'
# awq相关配置
# awq相关配置
try
:
try
:
...
...
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