From 722f8ebe9cae73e781116a52152f9ad6413d7adf Mon Sep 17 00:00:00 2001 From: Bryce Meyer Date: Sat, 28 Dec 2024 00:13:05 +0100 Subject: [PATCH] added test again --- tests/integration/test_match_huggingface.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/integration/test_match_huggingface.py b/tests/integration/test_match_huggingface.py index ef9d39e01..95ac80b35 100644 --- a/tests/integration/test_match_huggingface.py +++ b/tests/integration/test_match_huggingface.py @@ -42,4 +42,4 @@ def test_compare_huggingface_attention_match_local_implementation(self, model_na ) hf_out, _, _ = hf_model.transformer.h[layer_n].attn(hidden_states=input) - # assert torch.sum(tl_out == hf_out) == math.prod(tl_out.shape) + assert torch.sum(tl_out == hf_out) == math.prod(tl_out.shape)