[tokenizer] Support pluggable tokenizer worker class in multi-tokenizer mode (#30630)
Co-authored-by: tanujtiwari1998 <168470992+tanujtiwari1998@users.noreply.github.com>
This commit is contained in:
co-authored by
tanujtiwari1998
parent
77b7698cad
commit
cfc66e05c5
@@ -146,6 +146,7 @@ from sglang.srt.managers.multi_tokenizer_mixin import (
|
||||
MultiTokenizerRouter,
|
||||
TokenizerWorker,
|
||||
get_main_process_id,
|
||||
get_tokenizer_worker_class,
|
||||
read_from_shared_memory,
|
||||
write_data_for_multi_tokenizer,
|
||||
)
|
||||
@@ -236,7 +237,8 @@ async def init_multi_tokenizer() -> ServerArgs:
|
||||
)
|
||||
|
||||
# Launch multi-tokenizer manager process
|
||||
tokenizer_manager = TokenizerWorker(server_args, port_args)
|
||||
tokenizer_worker_class = get_tokenizer_worker_class(server_args)
|
||||
tokenizer_manager = tokenizer_worker_class(server_args, port_args)
|
||||
template_manager = TemplateManager()
|
||||
template_manager.initialize_templates(
|
||||
tokenizer_manager=tokenizer_manager,
|
||||
|
||||
Reference in New Issue
Block a user