transformers/tests/models/mpt
Mohamed Abu El-Nasr 0c2121f99b
Fix building alibi tensor when num_heads is not a power of 2 (#28380)
* Fix building alibi tensor when num_heads is not a power of 2

* Remove print function
2024-01-08 10:39:40 +01:00
..
__init__.py [`MPT`] Add MosaicML's `MPT` model to transformers (#24629) 2023-07-25 14:32:40 +02:00
test_modeling_mpt.py Fix building alibi tensor when num_heads is not a power of 2 (#28380) 2024-01-08 10:39:40 +01:00