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