[Mypy] Better fixes for the mypy issues in vllm/config (#37902)
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
This commit is contained in:
@@ -10,7 +10,6 @@ multimodal documents (text + images/videos).
|
||||
|
||||
from argparse import Namespace
|
||||
from collections.abc import Callable
|
||||
from dataclasses import asdict
|
||||
from pathlib import Path
|
||||
from typing import NamedTuple
|
||||
|
||||
@@ -125,7 +124,7 @@ def main(args: Namespace):
|
||||
model_request = model_example_map[args.model_name]()
|
||||
engine_args = model_request.engine_args
|
||||
|
||||
llm = LLM(**asdict(engine_args))
|
||||
llm = LLM.from_engine_args(engine_args)
|
||||
|
||||
print("Query: string & Document: string")
|
||||
outputs = llm.score(query, document)
|
||||
|
||||
Reference in New Issue
Block a user