tests/test_modeling_common.py · 5c6eca71a983bae2589eed01e5c04fcf88ba5690 · chenpangpang / transformers

fix `AutoModel.from_pretrained(..., torch_dtype=...)` (#13209) · 5c6eca71

Stas Bekman authored Aug 24, 2021

* fix AutoModel.from_pretrained(..., torch_dtype=...)

* fix to_diff_dict

* add better test

* torch is not always available when a model has self.torch_dtype

5c6eca71

test_modeling_common.py 75.8 KB

Replace test_modeling_common.py