File tree Expand file tree Collapse file tree 1 file changed +4
-1
lines changed Expand file tree Collapse file tree 1 file changed +4
-1
lines changed Original file line number Diff line number Diff line change @@ -455,7 +455,6 @@ def __init__(self,
455
455
self .enable_prompt_adapter = (self .runner .prompt_adapter_config
456
456
is not None )
457
457
self .multi_modal_input_mapper = self .runner .multi_modal_input_mapper
458
- self .decode_only = True
459
458
460
459
# Attention metadata inputs.
461
460
if self .attn_backend is not None :
@@ -477,6 +476,10 @@ def prepare(self,
477
476
finished_requests_ids : Optional [List [str ]] = None ) -> None :
478
477
self .finished_requests_ids = finished_requests_ids
479
478
479
+ # if the current batch is decode-only.
480
+ # will be set to False if there is any non-decode request.
481
+ self .decode_only = True
482
+
480
483
# Intermediate data (data in CPU before going to GPU) for
481
484
# the current sequence group.
482
485
self .inter_data_list : List [
You can’t perform that action at this time.
0 commit comments