Add more refactored openai test & in CI (#7284)
This commit is contained in:
@@ -36,7 +36,7 @@ from fastapi.middleware.cors import CORSMiddleware
|
||||
from fastapi.responses import Response
|
||||
|
||||
from sglang.srt.disaggregation.utils import (
|
||||
FakeBootstrapHost,
|
||||
FAKE_BOOTSTRAP_HOST,
|
||||
register_disaggregation_server,
|
||||
)
|
||||
from sglang.srt.entrypoints.engine import Engine, _launch_subprocesses
|
||||
@@ -265,7 +265,7 @@ def _wait_and_warmup(
|
||||
"max_new_tokens": 8,
|
||||
"ignore_eos": True,
|
||||
},
|
||||
"bootstrap_host": [FakeBootstrapHost] * server_args.dp_size,
|
||||
"bootstrap_host": [FAKE_BOOTSTRAP_HOST] * server_args.dp_size,
|
||||
# This is a hack to ensure fake transfer is enabled during prefill warmup
|
||||
# ensure each dp rank has a unique bootstrap_room during prefill warmup
|
||||
"bootstrap_room": [
|
||||
|
||||
Reference in New Issue
Block a user