Skip to content

Commit f5a7386

Browse files
Bounty-hunterdengyunyang
andauthored
bug fix for text request (vllm-project#52)
Co-authored-by: dengyunyang <[email protected]>
1 parent 8c8f477 commit f5a7386

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

vllm/v1/core/sched/scheduler.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -317,7 +317,8 @@ def schedule(self) -> SchedulerOutput:
317317
for i in encoder_inputs_to_schedule:
318318
self.encoder_cache_manager.allocate(request, i)
319319
encoder_budget = new_encoder_budget
320-
if (self.ec_connector is not None and external_load_encoder_input):
320+
if (self.ec_connector is not None and request.has_encoder_inputs
321+
and external_load_encoder_input):
321322
for i in external_load_encoder_input:
322323
self.encoder_cache_manager.allocate(request, i)
323324
self.ec_connector.update_state_after_alloc(request, i)
@@ -503,6 +504,7 @@ def schedule(self) -> SchedulerOutput:
503504
encoder_budget = new_encoder_budget
504505
# Allocate for external load encoder cache
505506
if (self.ec_connector is not None
507+
and request.has_encoder_inputs
506508
and external_load_encoder_input):
507509
for i in external_load_encoder_input:
508510
self.encoder_cache_manager.allocate(request, i)

0 commit comments

Comments
 (0)