tests/test_modeling_common.py · 7bc88c0511f7e455671872e332f7e40ccd25c8e7 · chenpangpang / transformers

"vscode:/vscode.git/clone" did not exist on "d8222be57ecc92646433a0b4b9025168894062bb"

[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