tests/models/bert/test_modeling_bert.py · 50378cbf6c1fd8717a74b36c352f57f9a73e7282 · chenpangpang / transformers

device agnostic models testing (#27146) · 50378cbf

Hz, Ji authored Nov 01, 2023

* device agnostic models testing

* add decorator `require_torch_fp16`

* make style

* apply review suggestion

* Oops, the fp16 decorator was misused

50378cbf

test_modeling_bert.py 26.4 KB

Replace test_modeling_bert.py