Skip to content

Commit e06d926

Browse files
committed
add humaneval to available datasets
Signed-off-by: Benjamin Chislett <bchislett@nvidia.com>
1 parent 3bbf4c0 commit e06d926

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

examples/specdec_bench/run.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@
3535
"SPECBENCH_MEDUSA": models.SpecBenchMedusaModel,
3636
}
3737
datasets_available = {
38+
"humaneval": datasets.HumanEval,
3839
"mtbench": datasets.MTBench,
3940
"random": datasets.RandomToken,
4041
"specbench": datasets.SpecBench,

0 commit comments

Comments
 (0)