fix red check-copies (#33964)

2024-10-04 22:45:37 +02:00
parent f319ba16fa
commit f92d354823
4 changed files with 3 additions and 57 deletions
--- a/tests/models/phimoe/test_modeling_phimoe.py
+++ b/tests/models/phimoe/test_modeling_phimoe.py
@@ -154,7 +154,7 @@ class PhimoeModelTester:

        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: