tests/test_modeling_common.py · 75343de938f2b57dd6b6f5bd16354900b8fbbc7f · chenpangpang / transformers

[modeling_utils] torch_dtype/auto floating dtype fixes (#17614) · 75343de9

Stas Bekman authored Jun 09, 2022



* [modeling_utils] torch_dtype/auto fixes

* add test

* apply suggestions

* add missing fallback

* Renaming things

* Use for else
Co-authored-by: Sylvain Gugger <Sylvain.gugger@gmail.com>

75343de9

test_modeling_common.py 127 KB

Replace test_modeling_common.py