File tree Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Original file line number Diff line number Diff 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 )
You can’t perform that action at this time.
0 commit comments