"lib/bindings/python/vscode:/vscode.git/clone" did not exist on "51b4cd7e880d7bd7a1bbec344eadbe07e7546a14"
Unverified Commit c40f3525 authored by jh-nv's avatar jh-nv Committed by GitHub
Browse files

chore: use config.modality directly for multimodal init and remove incorrect fallback value (#6471)

parent 038b50d2
......@@ -166,7 +166,6 @@ async def init_llm_worker(
)
kv_connector_config = build_kv_connector_config(config)
modality = getattr(config, "modality", None) or "text"
arg_map = {
"model": model_path,
"scheduler_config": scheduler_config,
......@@ -295,7 +294,7 @@ async def init_llm_worker(
# This overrides the skip_tokenizer_init=True set earlier
engine_args["skip_tokenizer_init"] = False
if modality == Modality.MULTIMODAL:
if config.modality == Modality.MULTIMODAL:
engine_args["skip_tokenizer_init"] = False
model_config = AutoConfig.from_pretrained(config.model, trust_remote_code=True)
multimodal_processor = MultimodalRequestProcessor(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment