tests/test_modeling_common.py · cd6bd0af3435ab731f7ed1147def5fae3ffc941e · chenpangpang / transformers

Benjamin Warner authored May 20, 2024

* add torch.compile dynamic support

* Add SDPA dynamic shapes compile test & improve SDPA comment

* comment consistency

cd6bd0af

test_modeling_common.py 211 KB