[Core] Remove tokenizer group in vLLM (#24078)
Signed-off-by: Zhuohan Li <zhuohan123@gmail.com>
This commit is contained in:
@@ -390,11 +390,8 @@ class _AsyncLLMEngine(LLMEngine):
|
||||
"""Stop the remote worker execution loop."""
|
||||
await self.model_executor.stop_remote_worker_execution_loop_async()
|
||||
|
||||
async def get_tokenizer_async(self,
|
||||
lora_request: Optional[LoRARequest] = None
|
||||
) -> AnyTokenizer:
|
||||
return await (
|
||||
self.get_tokenizer_group().get_lora_tokenizer_async(lora_request))
|
||||
async def get_tokenizer_async(self) -> AnyTokenizer:
|
||||
return self.get_tokenizer()
|
||||
|
||||
async def add_request_async(
|
||||
self,
|
||||
@@ -435,7 +432,6 @@ class _AsyncLLMEngine(LLMEngine):
|
||||
|
||||
processed_inputs = await self.input_preprocessor.preprocess_async(
|
||||
prompt,
|
||||
lora_request=lora_request,
|
||||
tokenization_kwargs=tokenization_kwargs,
|
||||
)
|
||||
|
||||
@@ -614,11 +610,8 @@ class AsyncLLMEngine(EngineClient):
|
||||
async def get_input_preprocessor(self) -> InputPreprocessor:
|
||||
return self.engine.input_preprocessor
|
||||
|
||||
async def get_tokenizer(
|
||||
self,
|
||||
lora_request: Optional[LoRARequest] = None,
|
||||
) -> AnyTokenizer:
|
||||
return await self.engine.get_tokenizer_async(lora_request)
|
||||
async def get_tokenizer(self) -> AnyTokenizer:
|
||||
return self.engine.get_tokenizer()
|
||||
|
||||
def start_background_loop(self) -> None:
|
||||
"""Start the background loop."""
|
||||
|
||||
Reference in New Issue
Block a user