tests/models/falcon/test_modeling_falcon.py · 1897874edcba58e0e585305ca71f6ba727315219 · chenpangpang / transformers

Fix falcon with SDPA, alibi but no passed mask (#30123) · 1897874e

fxmarty authored Apr 08, 2024



* fix falcon without attention_mask & alibi

* add test

* Update tests/models/falcon/test_modeling_falcon.py
Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

---------
Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

1897874e

test_modeling_falcon.py 29 KB

Replace test_modeling_falcon.py