transformers/tests/models/opt
Patrick von Platen e8714c0307
[OPT] Run test in lower precision on GPU (#17353)
* [OPT] Run test only in half precision

* up

* up

* up

* up

* finish

* fix on GPU

* Update tests/models/opt/test_modeling_opt.py
2022-05-19 22:15:36 +02:00
..
__init__.py Add OPT (#17088) 2022-05-12 12:24:35 +02:00
test_modeling_opt.py [OPT] Run test in lower precision on GPU (#17353) 2022-05-19 22:15:36 +02:00