Skip to content

Commit

Permalink
TST Enable LoftQ 8bit tests (#1279)
Browse files Browse the repository at this point in the history
Due to PR #1276, the bug that prevented use of LoftQ with 8bit
quantization has now been fixed. Therefore, the tests no longer need to
be skipped.
  • Loading branch information
BenjaminBossan authored Dec 18, 2023
1 parent 119de17 commit 482a2a6
Showing 1 changed file with 0 additions and 10 deletions.
10 changes: 0 additions & 10 deletions tests/test_gpu_examples.py
Original file line number Diff line number Diff line change
Expand Up @@ -1060,11 +1060,6 @@ def test_bloomz_loftq_4bit_iter_5(self, device):

@parameterized.expand(["cuda", "cpu"])
def test_bloomz_loftq_8bit(self, device):
# this currently does not work:
# https://github.com/huggingface/peft/pull/1150#issuecomment-1838891499
if True: # TODO: remove as soon as the issue is fixed
return

# Same test as test_bloomz_loftq_4bit but with 8 bits.
mae_quantized, mse_quantized, mae_loftq, mse_loftq = self.get_errors(bits=8, device=device)

Expand All @@ -1080,11 +1075,6 @@ def test_bloomz_loftq_8bit(self, device):

@parameterized.expand(["cuda", "cpu"])
def test_bloomz_loftq_8bit_iter_5(self, device):
# this currently does not work:
# https://github.com/huggingface/peft/pull/1150#issuecomment-1838891499
if True: # TODO: remove as soon as the issue is fixed
return

# Same test as test_bloomz_loftq_4bit_iter_5 but with 8 bits.
mae_quantized, mse_quantized, mae_loftq, mse_loftq = self.get_errors(bits=8, loftq_iter=5, device=device)

Expand Down

0 comments on commit 482a2a6

Please sign in to comment.