[lora] Exclude finished requests from running_loras (#27597)
This commit is contained in:
@@ -2641,7 +2641,9 @@ class Scheduler(
|
|||||||
self.chunked_req = adder.add_chunked_req(self.chunked_req)
|
self.chunked_req = adder.add_chunked_req(self.chunked_req)
|
||||||
|
|
||||||
if self.enable_lora:
|
if self.enable_lora:
|
||||||
running_loras = {req.lora_id for req in self.running_batch.reqs}
|
running_loras = {
|
||||||
|
req.lora_id for req in self.running_batch.reqs if not req.finished()
|
||||||
|
}
|
||||||
# Account for LoRAs that are already loaded in the adder, such as chunked requests
|
# Account for LoRAs that are already loaded in the adder, such as chunked requests
|
||||||
running_loras.update(req.lora_id for req in adder.can_run_list)
|
running_loras.update(req.lora_id for req in adder.can_run_list)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user