Skip to content

Commit

Permalink
Update tests/test_modeling_common.py
Browse files Browse the repository at this point in the history
Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
  • Loading branch information
2 people authored and Sebastien Ehrhardt committed May 14, 2024
1 parent 99f4c67 commit 48e137c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/test_modeling_common.py
Original file line number Diff line number Diff line change
Expand Up @@ -3864,7 +3864,7 @@ def get_mean_reldiff(failcase, x, ref, atol, rtol):
):
dummy_mask = torch.ones((self.model_tester.num_masks,))

# In case of additional token (like class) we define a custome `mask_length`
# In case of additional token (like class) we define a custom `mask_length`
if hasattr(self.model_tester, "mask_length"):
dummy_mask = torch.cat(
[
Expand Down

0 comments on commit 48e137c

Please sign in to comment.