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

[ci][test] exclude model download time in server start time (#7834)

Browse Source
This commit is contained in:
youkaichao 2024-08-24 01:03:27 -07:00 committed by GitHub
parent 7d9ffa2ae1
commit ea9fa160e3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/utils.py
View File

@ -11,6 +11,7 @@ from typing import Any, Callable, Dict, List, Optional
import openai import openai
import requests import requests
from huggingface_hub import snapshot_download
from transformers import AutoTokenizer from transformers import AutoTokenizer
from typing_extensions import ParamSpec from typing_extensions import ParamSpec
@ -64,6 +65,10 @@ class RemoteOpenAIServer:
env_dict: Optional[Dict[str, str]] = None, env_dict: Optional[Dict[str, str]] = None,
auto_port: bool = True, auto_port: bool = True,
max_wait_seconds: Optional[float] = None) -> None: max_wait_seconds: Optional[float] = None) -> None:
if not model.startswith("/"):
# download the model if it's not a local path
# to exclude the model download time from the server start time
model = snapshot_download(model)
if auto_port: if auto_port:
if "-p" in cli_args or "--port" in cli_args: if "-p" in cli_args or "--port" in cli_args:
raise ValueError("You have manually specified the port" raise ValueError("You have manually specified the port"