transformers/tests/models/xglm
Gabriele Sarti 9d732fd2dd
XGLM - Fix Softmax NaNs when using FP16 (#18057)
* fix fp16 for xglm

* Removed misleading comment

* Fix undefined variable

Co-authored-by: Gabriele Sarti <gsarti@amazon.com>
Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
Co-authored-by: Younes Belkada <49240599+younesbelkada@users.noreply.github.com>
2022-09-29 10:42:07 +02:00
..
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_modeling_flax_xglm.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_modeling_tf_xglm.py Add TF implementation of XGLMModel (#16543) 2022-08-24 10:51:05 +01:00
test_modeling_xglm.py XGLM - Fix Softmax NaNs when using FP16 (#18057) 2022-09-29 10:42:07 +02:00
test_tokenization_xglm.py Black preview (#17217) 2022-05-12 16:25:55 -04:00