simcop2387/AutoGPTQ
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 1 Wiki Activity

fix atol

Browse source
This commit is contained in:
Felix Marty 2023-07-31 16:46:15 +00:00
parent 339c57a902
commit 7ce182888c
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tests/test_q4.py
View file

@ -186,7 +186,7 @@ class TestsQ4Exllama(unittest.TestCase):
reference = self.REFERENCE.to(device)
self.assertTrue(torch.allclose(res, reference, rtol=3e-5, atol=1e-2), get_diff(res, reference))
self.assertTrue(torch.allclose(res, reference, rtol=3e-5, atol=2e-2), get_diff(res, reference))
def test_generation_no_act_order(self):
prompt = "I am in Paris and"