Add scaled_dot_product_attention support for decoder models #853
add gpt2
fe9edd36
add files
a88c9465
refactor tests
e575f22d
support gpt2, better tests
f171271a
fxmarty
marked this pull request as draft 2 years ago
fix
d1a30c8d
fxmarty
marked this pull request as ready for review 2 years ago
more models
00a04df0
fix gpt neo
5ef8972d
fxmarty
changed the title Add scaled_dot_product_attention support for gpt2 Add scaled_dot_product_attention support for decoder models 2 years ago
small fixes
652e1f28
add opt support
72cb776a
fix tests
245098bc
add comment
c466be5b
fix mock
6747a41a
fix uninstall
0ee402a6
size
e955b6f9
last fix
bab7a039
fxmarty
force pushed
from
f876e596
to
bab7a039
2 years ago
fxmarty
merged
6df75975
into main 2 years ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub