tests/test_modeling_common.py · 5a4a76edb7ac6bbc764392e89adc11adda91f3e5 · chenpangpang / transformers

Modify resize_token_embeddings to ensure output type is same as input (#31979) · 5a4a76ed

bayllama authored Jul 23, 2024



* Change resize_token_embeddings to make it return same Class that is passed to it

* Add explanatory comment as requested in review

* Add explanatory comments for add resizing function in lxmert

* Add comment for padding_idx and moving _resize_bias in lxmert to LxmertForPreTraining

---------
Co-authored-by: Prashanth Sateesh <prasatee@Prashanths-MBP.attlocal.net>
Co-authored-by: Prashanth Sateesh <prasatee@Prashanths-MacBook-Pro.local>

5a4a76ed

test_modeling_common.py 215 KB

Replace test_modeling_common.py