Skip to content

Commit

Permalink
attempts to fix lmeval failing tests
Browse files Browse the repository at this point in the history
Signed-off-by: Brian Dellabetta <[email protected]>
  • Loading branch information
brian-dellabetta committed Mar 4, 2025
1 parent 7bb517f commit ca5a6ba
Show file tree
Hide file tree
Showing 4 changed files with 8 additions and 8 deletions.
9 changes: 6 additions & 3 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -60,9 +60,12 @@
"datasets",
"accelerate>=0.20.3,!=1.1.0",
"pynvml",
"compressed-tensors"
if version_info.build_type == "release"
else "compressed-tensors-nightly",
"pillow",
(
"compressed-tensors"
if version_info.build_type == "release"
else "compressed-tensors-nightly"
),
],
extras_require={
"dev": [
Expand Down
3 changes: 1 addition & 2 deletions tests/lmeval/configs/vl_fp8_dynamic_per_token.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,6 @@ lmeval:
convert_img_format: True
task: mmmu_val_economics
num_fewshot: 0
limit: 1000
batch_size: 8
metrics:
acc,none: 0.333
acc,none: 0.266
1 change: 0 additions & 1 deletion tests/lmeval/configs/vl_int8_w8a8_dynamic_per_token.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,6 @@ lmeval:
convert_img_format: True
task: mmmu_val_economics
num_fewshot: 0
limit: 1000
metrics:
acc,none: 0.233
batch_size: 8
3 changes: 1 addition & 2 deletions tests/lmeval/configs/vl_w4a16_actorder_weight.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,6 @@ lmeval:
convert_img_format: True
task: mmmu_val_economics
num_fewshot: 0
limit: 1000
metrics:
acc,none: 0.4
acc,none: 0.433
batch_size: 4

0 comments on commit ca5a6ba

Please sign in to comment.