Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 2 additions & 3 deletions benchmark/lora/lora_bench.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,16 +24,15 @@
from datetime import datetime
from typing import Any, Dict, List, Optional, Tuple

import aiohttp
import numpy as np
from launch_server import LORA_PATH, NUM_LORAS
from tqdm.asyncio import tqdm
from transformers import PreTrainedTokenizerBase

from sglang.bench_serving import (
AIOHTTP_TIMEOUT,
RequestFuncInput,
RequestFuncOutput,
_create_bench_client_session,
calculate_metrics,
get_request,
get_tokenizer,
Expand All @@ -56,7 +55,7 @@ async def async_request_openai_completions(

prompt = request_func_input.prompt

async with aiohttp.ClientSession(timeout=AIOHTTP_TIMEOUT) as session:
async with _create_bench_client_session() as session:
# payload = {
# "model": request_func_input.model,
# "prompt": prompt,
Expand Down
Loading