Skip to content

Commit

Permalink
Merge branch 'vllm-project-main' into image_bench-on-refactor
Browse files Browse the repository at this point in the history
  • Loading branch information
Linkun Chen committed Nov 4, 2024
2 parents 32ae85a + 8dd782e commit 6da8645
Showing 1 changed file with 0 additions and 18 deletions.
18 changes: 0 additions & 18 deletions benchmarks/benchmark_throughput.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,6 @@
import random
import time
from typing import List, Optional
from typing import List, Optional

import torch
import uvloop
Expand All @@ -23,23 +22,6 @@
from vllm.utils import FlexibleArgumentParser, merge_async_iterators


@dataclasses.dataclass
class SampleRequest:
"""A class representing a single inference request for benchmarking.
Attributes:
prompt: The input text prompt for the model.
multi_modal_data: Optional dictionary containing multi-modal data (e.g.
images).
prompt_len: The length of the prompt in tokens.
expected_output_len: The expected length of the output in tokens.
"""
prompt: str
prompt_len: int
expected_output_len: int
multi_modal_data: Optional[MultiModalDataDict] = None


@dataclasses.dataclass
class SampleRequest:
"""A class representing a single inference request for benchmarking.
Expand Down

0 comments on commit 6da8645

Please sign in to comment.