Stas Bekman bbbc5c15d4 [AutoModel] fix torch_dtype=auto in from_pretrained (#23379) ...

* [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>

2023-05-16 10:21:42 -07:00

161 KiB

Executable File

Raw Blame History

View Raw