SIGN IN SIGN UP
hpcaitech / ColossalAI UNCLAIMED

Making large AI models cheaper, faster and more accessible

0 0 38 Python

[shardformer]fix flash attention, when mask is casual, just don't unpad it (#5084)

* fix flash attn

* fix

fix
F
flybird11111 committed
aae496631c4acac03d8707216958184290a018fd
Parent: 75af66c
Committed by GitHub <noreply@github.com> on 11/22/2023, 8:00:07 AM