squall/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

[Bugfix][Hardware][AMD][Frontend] add quantization param to embedding checking method (#7513)

Browse Source
This commit is contained in:
Gordon Wong 2024-08-17 00:05:18 +08:00 committed by GitHub
parent 6fc5b0f249
commit 0e39a33c6d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
vllm/entrypoints/openai/api_server.py
View File

@ -60,11 +60,13 @@ logger = init_logger('vllm.entrypoints.openai.api_server')
_running_tasks: Set[asyncio.Task] = set() _running_tasks: Set[asyncio.Task] = set()
def model_is_embedding(model_name: str, trust_remote_code: bool) -> bool: def model_is_embedding(model_name: str, trust_remote_code: bool,
quantization: str) -> bool:
return ModelConfig(model=model_name, return ModelConfig(model=model_name,
tokenizer=model_name, tokenizer=model_name,
tokenizer_mode="auto", tokenizer_mode="auto",
trust_remote_code=trust_remote_code, trust_remote_code=trust_remote_code,
quantization=quantization,
seed=0, seed=0,
dtype="auto").embedding_mode dtype="auto").embedding_mode
@ -97,7 +99,8 @@ async def build_async_engine_client(args) -> AsyncIterator[AsyncEngineClient]:
# If manually triggered or embedding model, use AsyncLLMEngine in process. # If manually triggered or embedding model, use AsyncLLMEngine in process.
# TODO: support embedding model via RPC. # TODO: support embedding model via RPC.
if (model_is_embedding(args.model, args.trust_remote_code) if (model_is_embedding(args.model, args.trust_remote_code,
args.quantization)
or args.disable_frontend_multiprocessing): or args.disable_frontend_multiprocessing):
async_engine_client = AsyncLLMEngine.from_engine_args( async_engine_client = AsyncLLMEngine.from_engine_args(
engine_args, usage_context=UsageContext.OPENAI_API_SERVER) engine_args, usage_context=UsageContext.OPENAI_API_SERVER)