typo
This commit is contained in:
parent
f1886e5be1
commit
ff721107b9
@ -26,5 +26,3 @@ def get_prefix_causal_mask_from_padding_mask(seq_len:int, prefix_mask, att_heads
|
|||||||
|
|
||||||
|
|
||||||
|
|
||||||
print(get_causal_attention_mask_with_prefix(10,3))
|
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user