Generate tests: modality-agnostic input preparation (#33685)

2024-10-03 14:01:24 +01:00
parent f2bf4fcf3d
commit d29738f5b4
34 changed files with 241 additions and 906 deletions
--- a/tests/models/starcoder2/test_modeling_starcoder2.py
+++ b/tests/models/starcoder2/test_modeling_starcoder2.py
@@ -107,7 +107,7 @@ class Starcoder2ModelTester:

        input_mask = None
        if self.use_input_mask:
-            input_mask = torch.tril(torch.ones(self.batch_size, self.seq_length)).to(torch_device)
+            input_mask = torch.tril(torch.ones_like(input_ids).to(torch_device))

        token_type_ids = None
        if self.use_token_type_ids: