Generate: add test to check KV format (#23403)

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
This commit is contained in:
Joao Gante
2023-05-16 19:28:19 +01:00
committed by GitHub
parent 9cf4a8b456
commit 918a06e25d
4 changed files with 84 additions and 1 deletions

View File

@@ -831,8 +831,12 @@ class ReformerLSHAttnModelTest(
[expected_shape] * len(iter_hidden_states),
)
@unittest.skip("Fails because the sequence length is not a multiple of 4")
def test_problem_types(self):
# Fails because the sequence length is not a multiple of 4
pass
@unittest.skip("Fails because the sequence length is not a multiple of 4")
def test_past_key_values_format(self):
pass