mirror of
https://github.com/huggingface/peft.git
synced 2025-10-20 15:33:48 +08:00
TST FIX Failing AutoAWQ test with torch 2.8 (#2752)
There is a failing AWQ test since torch 2.6 which is marked as xfail for torch=2.7. However, now torch 2.8 is out and the test is still failing. Therefore, the xfail now checks for torch>=2.7. As AWQ is no longer being maintained, we should expect this situation to deteriorate over time and eventually we'll have to remove it. But for the time being, it still appears to mostly work, so I suggest we leave it as is.
This commit is contained in:
@ -3712,6 +3712,13 @@ class PeftHqqGPUTests(unittest.TestCase):
|
||||
class PeftAwqGPUTests(unittest.TestCase):
|
||||
r"""
|
||||
Awq + peft tests
|
||||
|
||||
Note that AWQ is no longer being maintained:
|
||||
|
||||
https://github.com/casper-hansen/AutoAWQ/blob/88e4c76b20755db275574e6a03c83c84ba3bece5/README.md
|
||||
|
||||
It is therefore expected that more tests will start failing in the future. If this happens, remove AWQ support from
|
||||
PEFT.
|
||||
"""
|
||||
|
||||
def setUp(self):
|
||||
@ -3792,8 +3799,8 @@ class PeftAwqGPUTests(unittest.TestCase):
|
||||
# TODO remove marker if/once issue is resolved, most likely requiring a fix in AutoAWQ:
|
||||
# https://github.com/casper-hansen/AutoAWQ/issues/754
|
||||
@pytest.mark.xfail(
|
||||
condition=is_torch_version("==", "2.7.0") or is_torch_version("==", "2.7.1"),
|
||||
reason="Multi-GPU test currently not working with AutoAWQ and PyTorch 2.7",
|
||||
condition=is_torch_version(">=", "2.7.0"),
|
||||
reason="Multi-GPU test currently not working with AutoAWQ and PyTorch 2.7+",
|
||||
strict=True,
|
||||
)
|
||||
@require_torch_multi_accelerator
|
||||
|
Reference in New Issue
Block a user