LLM/airllm
1
0
Fork 0
mirror of https://github.com/0xSojalSec/airllm.git synced 2026-03-07 22:33:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix ut

Browse Source
This commit is contained in:
Yu Li
2023-12-01 16:31:43 -06:00
parent e1c1aa6317
commit bedc395df8
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
air_llm/tests/test_compression.py
View File

@@ -34,7 +34,7 @@ class TestCompression(unittest.TestCase):
for k in aa.keys():
if compression is None:
self.assertAlmostEqual(aa[k], a_state_dict[k])
self.assertTrue(torch.eq(aa[k], a_state_dict[k]))
else:
RMSE_loss = torch.sqrt(loss_fn(aa[k], a_state_dict[k]))
self.assertLess(RMSE_loss.detach().numpy()[0], 0.5)