tests/test_modeling_utils.py · 76a33a10923ccc1074917f6b6a1e719e626b7dc9 · chenpangpang / transformers

Llama: partial 4d masks (#29731) · 4294f0c3

Joao Gante authored Mar 19, 2024



* partial 4d masks

* Apply suggestions from code review
Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

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

4294f0c3

test_modeling_utils.py 96 KB

Replace test_modeling_utils.py