diff --git a/python/sglang/srt/managers/tp_worker.py b/python/sglang/srt/managers/tp_worker.py index 0d561aab7..1b8615e30 100644 --- a/python/sglang/srt/managers/tp_worker.py +++ b/python/sglang/srt/managers/tp_worker.py @@ -192,7 +192,8 @@ class BaseTpWorker(ABC): return result def can_run_lora_batch(self, lora_ids: list[str]) -> bool: - return self.model_runner.lora_manager.validate_lora_batch(lora_ids) + lora_ids_set = set(lora_ids) if isinstance(lora_ids, list) else lora_ids + return self.model_runner.lora_manager.validate_lora_batch(lora_ids_set) def forward_batch_embedding(self, model_worker_batch: ModelWorkerBatch): forward_batch = ForwardBatch.init_new(model_worker_batch, self.model_runner)