src/transformers/modeling_utils.py · 123cce6ffcafe6ecece7eb9914395d2ebcc98a48 · chenpangpang / transformers

[modeling_utils] respect original dtype in _get_resized_lm_head (#14181) · 123cce6f

Stas Bekman authored Oct 27, 2021



* respect dtype in _get_resized_lm_head

* Update src/transformers/modeling_utils.py
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>

* consistency
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>

123cce6f

modeling_utils.py 111 KB

Replace modeling_utils.py