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
7f881746
Commit
7f881746
authored
Feb 22, 2025
by
zhuwenwen
Browse files
set enforce eager
parent
0a89b8a3
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
4 additions
and
2 deletions
+4
-2
vllm/config.py
vllm/config.py
+2
-1
vllm/engine/arg_utils.py
vllm/engine/arg_utils.py
+2
-1
No files found.
vllm/config.py
View file @
7f881746
...
...
@@ -322,7 +322,8 @@ class ModelConfig:
# Set enforce_eager to False if the value is unset.
if
self
.
enforce_eager
is
None
:
self
.
enforce_eager
=
False
# self.enforce_eager = False
self
.
enforce_eager
=
True
sliding_window
=
getattr
(
self
.
hf_text_config
,
"sliding_window"
,
None
)
has_interleaved_attention
=
(
sliding_window
is
not
None
)
and
(
...
...
vllm/engine/arg_utils.py
View file @
7f881746
...
...
@@ -564,7 +564,8 @@ class EngineArgs:
'This should be a JSON string that will be '
'parsed into a dictionary.'
)
parser
.
add_argument
(
'--enforce-eager'
,
action
=
'store_true'
,
# action='store_true',
default
=
True
,
help
=
'Always use eager-mode PyTorch. If False, '
'will use eager mode and CUDA graph in hybrid '
'for maximal performance and flexibility.'
)
...
...
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