[Tokenizer] Add tokenizer mode (#298)

This commit is contained in:
Woosuk Kwon
2023-06-28 14:19:22 -07:00
committed by GitHub
parent 425040d4c1
commit 998d9d1509
6 changed files with 38 additions and 5 deletions

View File

@@ -61,6 +61,7 @@ class LLMEngine:
"Initializing an LLM engine with config: "
f"model={model_config.model!r}, "
f"tokenizer={model_config.tokenizer!r}, "
f"tokenizer_mode={model_config.tokenizer_mode}, "
f"dtype={model_config.dtype}, "
f"use_dummy_weights={model_config.use_dummy_weights}, "
f"download_dir={model_config.download_dir!r}, "
@@ -77,7 +78,8 @@ class LLMEngine:
self.log_stats = log_stats
self._verify_args()
self.tokenizer = get_tokenizer(model_config.tokenizer)
self.tokenizer = get_tokenizer(model_config.tokenizer,
model_config.tokenizer_mode)
self.seq_counter = Counter()
# Create the parallel GPU workers.