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