Skip to content

Commit c5d77b4

Browse files
committed
Remove piqa from tests
1 parent 2ba842b commit c5d77b4

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

requirements/requirements-llm.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
accelerate
22
datasets
33
gguf==0.17.0
4-
lighteval; python_version >= "3.10"
4+
lighteval[math]; python_version >= "3.10"
55
lm_eval
66
onnx
77
onnxruntime<1.21

tests/brevitas_examples/test_llm.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -385,8 +385,7 @@ def quantize_llm_assert_args(args, extra_args=None):
385385
"leaderboard|arc:challenge|0|0",
386386
"leaderboard|winogrande|0|0",
387387
"lighteval|arc:easy|0|0",
388-
"leaderboard|hellaswag|0|0",
389-
"lighteval|piqa|0|0",],
388+
"leaderboard|hellaswag|0|0",],
390389
"few_shot_zeroshot": True,
391390
"imports": ["lighteval"],
392391
"all_acc": 0.375,},])

0 commit comments

Comments
 (0)