Skip to content

Commit 6a7046c

Browse files
TST AQLM test no longer x-fails (#2506)
There has been an AQLM release with the fix to the issue that originally broke the test.
1 parent f7cda1f commit 6a7046c

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

tests/test_gpu_examples.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2880,8 +2880,6 @@ def _check_inference_finite(self, model, batch):
28802880
model.train(training)
28812881

28822882
@pytest.mark.single_gpu_tests
2883-
# see https://github.com/Vahe1994/AQLM/pull/139
2884-
@pytest.mark.xfail(reason="AQLM does not work with PyTorch 2.5 (yet)", strict=True, raises=AttributeError)
28852883
def test_causal_lm_training_aqlm(self):
28862884
r"""
28872885
Test the CausalLM training on a single GPU device. The test would simply fail if the adapters are not set

0 commit comments

Comments
 (0)