tests/test_modeling_common.py · bbbc5c15d4fbf87f74fc4e5caa78d94d3d73bbce · chenpangpang / transformers

[AutoModel] fix `torch_dtype=auto` in `from_pretrained` (#23379) · bbbc5c15

Stas Bekman authored May 16, 2023



* [automodel] fix torch_dtype=auto in from_pretrained

* add test

* fix logic

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

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

bbbc5c15

test_modeling_common.py 161 KB

Replace test_modeling_common.py