mirror of
https://github.com/huggingface/peft.git
synced 2025-10-20 15:33:48 +08:00
Compare commits
2 Commits
31989eab83
...
younesbelk
Author | SHA1 | Date | |
---|---|---|---|
b3253faa38 | |||
480d935e72 |
@ -1930,8 +1930,10 @@ class PeftAqlmGPUTests(unittest.TestCase):
|
||||
assert trainer.state.log_history[-1]["train_loss"] is not None
|
||||
|
||||
|
||||
# TODO: unskip the tests once https://github.com/casper-hansen/AutoAWQ/issues/466 is fixed
|
||||
@require_torch_gpu
|
||||
@require_auto_awq
|
||||
@pytest.mark.skip(reason="Needs https://github.com/casper-hansen/AutoAWQ/issues/466 to be fixed first")
|
||||
class PeftAwqGPUTests(unittest.TestCase):
|
||||
r"""
|
||||
Awq + peft tests
|
||||
|
Reference in New Issue
Block a user