Skip to content

Commit a6970a1

Browse files
authored
misc: fix accept_length (#6536)
1 parent a6ae3af commit a6970a1

File tree

1 file changed

+10
-7
lines changed

1 file changed

+10
-7
lines changed

python/sglang/bench_serving.py

Lines changed: 10 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1303,14 +1303,17 @@ async def limited_request_func(request_func_input, pbar):
13031303

13041304
if "sglang" in backend:
13051305
server_info = requests.get(base_url + "/get_server_info")
1306-
if pd_separated:
1307-
accept_length = server_info.json()["decode"][0]["internal_states"][0].get(
1308-
"avg_spec_accept_length", None
1309-
)
1306+
if server_info.status_code == 200:
1307+
if pd_separated:
1308+
accept_length = server_info.json()["decode"][0]["internal_states"][
1309+
0
1310+
].get("avg_spec_accept_length", None)
1311+
else:
1312+
accept_length = server_info.json()["internal_states"][0].get(
1313+
"avg_spec_accept_length", None
1314+
)
13101315
else:
1311-
accept_length = server_info.json()["internal_states"][0].get(
1312-
"avg_spec_accept_length", None
1313-
)
1316+
accept_length = None
13141317
else:
13151318
accept_length = None
13161319

0 commit comments

Comments
 (0)