tiny fix
This commit is contained in:
1 parent
636bb9c1e6
commit
738df47748
1 file changed
+1
-1
@@ -41,7 +41,7 @@ def prepare_4d_attention_mask(attention_mask_with_indices: "torch.Tensor", dtype
|
||||
[x, x, o, x, x, x],
|
||||
[x, x, o, o, x, x],
|
||||
[x, x, o, o, o, x],
|
||||
[x, x, o, x, x, x],
|
||||
[x, x, x, x, x, x],
|
||||
]
|
||||
]
|
||||
]
|
||||
|
||||
Reference in new issue
Block a user