Fix port conflicts in CI (#11497)
This commit is contained in:
@@ -76,6 +76,7 @@ decord = ["decord"]
|
||||
test = [
|
||||
"accelerate",
|
||||
"expecttest",
|
||||
"gguf",
|
||||
"jsonlines",
|
||||
"matplotlib",
|
||||
"pandas",
|
||||
@@ -85,10 +86,10 @@ test = [
|
||||
"tabulate",
|
||||
]
|
||||
tracing = [
|
||||
"opentelemetry-api",
|
||||
"opentelemetry-exporter-otlp",
|
||||
"opentelemetry-exporter-otlp-proto-grpc",
|
||||
"opentelemetry-sdk",
|
||||
"opentelemetry-api",
|
||||
"opentelemetry-exporter-otlp",
|
||||
"opentelemetry-exporter-otlp-proto-grpc",
|
||||
"opentelemetry-sdk",
|
||||
]
|
||||
all = ["sglang[test]", "sglang[decord]"]
|
||||
all_aarch64 = ["sglang[test]"]
|
||||
|
||||
@@ -135,11 +135,11 @@ def _use_cached_default_models(model_repo: str):
|
||||
|
||||
if is_in_ci():
|
||||
DEFAULT_PORT_FOR_SRT_TEST_RUNNER = (
|
||||
5000 + int(os.environ.get("CUDA_VISIBLE_DEVICES", "0")[0]) * 100
|
||||
10000 + int(os.environ.get("CUDA_VISIBLE_DEVICES", "0")[0]) * 1000
|
||||
)
|
||||
else:
|
||||
DEFAULT_PORT_FOR_SRT_TEST_RUNNER = (
|
||||
7000 + int(os.environ.get("CUDA_VISIBLE_DEVICES", "0")[0]) * 100
|
||||
20000 + int(os.environ.get("CUDA_VISIBLE_DEVICES", "0")[0]) * 1000
|
||||
)
|
||||
DEFAULT_URL_FOR_TEST = f"http://127.0.0.1:{DEFAULT_PORT_FOR_SRT_TEST_RUNNER + 1000}"
|
||||
|
||||
@@ -396,8 +396,6 @@ def _get_call_generate(args: argparse.Namespace):
|
||||
return partial(call_generate_vllm, url=f"{args.host}:{args.port}/generate")
|
||||
elif args.backend == "srt-raw":
|
||||
return partial(call_generate_srt_raw, url=f"{args.host}:{args.port}/generate")
|
||||
elif args.backend == "gserver":
|
||||
return partial(call_generate_gserver, url=f"{args.host}:{args.port}")
|
||||
elif args.backend == "outlines":
|
||||
return partial(call_generate_outlines, url=f"{args.host}:{args.port}/generate")
|
||||
elif args.backend == "guidance":
|
||||
@@ -521,7 +519,6 @@ def popen_launch_server(
|
||||
# Auto-detect device if needed
|
||||
if device == "auto":
|
||||
device = auto_config_device()
|
||||
print(f"Auto-configed device: {device}", flush=True)
|
||||
other_args = list(other_args)
|
||||
other_args += ["--device", str(device)]
|
||||
|
||||
|
||||
Reference in New Issue
Block a user