[Model] Support math-shepherd-mistral-7b-prm model (#9697)
Signed-off-by: Went-Liang <wenteng_liang@163.com>
This commit is contained in:
@@ -257,7 +257,8 @@ class LLMEngine:
|
||||
"num_scheduler_steps=%d, chunked_prefill_enabled=%s "
|
||||
"multi_step_stream_outputs=%s, enable_prefix_caching=%s, "
|
||||
"use_async_output_proc=%s, use_cached_outputs=%s, "
|
||||
"chat_template_text_format=%s, mm_processor_kwargs=%s)",
|
||||
"chat_template_text_format=%s, mm_processor_kwargs=%s, "
|
||||
"pooler_config=%r)",
|
||||
VLLM_VERSION,
|
||||
model_config.model,
|
||||
speculative_config,
|
||||
@@ -294,6 +295,7 @@ class LLMEngine:
|
||||
use_cached_outputs,
|
||||
model_config.chat_template_text_format,
|
||||
model_config.mm_processor_kwargs,
|
||||
model_config.pooler_config,
|
||||
)
|
||||
# TODO(woosuk): Print more configs in debug mode.
|
||||
self.model_config = model_config
|
||||
|
||||
Reference in New Issue
Block a user