Skip to content

Commit a3ab00f

Browse files
ekagra-ranjanywang96
authored andcommitted
[Spec Decode] Fix offline spec_decode.py (vllm-project#24257)
Signed-off-by: Ekagra Ranjan <[email protected]> Co-authored-by: Roger Wang <[email protected]> Signed-off-by: xuebwang-amd <[email protected]>
1 parent c79a509 commit a3ab00f

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

vllm/benchmarks/datasets.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1310,6 +1310,10 @@ def normalize(d: dict) -> dict[tuple[int, int, int], float]:
13101310

13111311

13121312
def get_samples(args, tokenizer) -> list[SampleRequest]:
1313+
1314+
if not hasattr(args, "request_id_prefix"):
1315+
args.request_id_prefix = ""
1316+
13131317
if args.dataset_name == "custom":
13141318
dataset = CustomDataset(dataset_path=args.dataset_path)
13151319
input_requests = dataset.sample(

0 commit comments

Comments
 (0)