transformers/tests/xglm
Suraj Patil 130b987880
[XGLM] run sampling test on CPU to be deterministic (#15892)
* run sampling test on CPU to be deterministic

* input_ids on CPU
2022-03-02 17:55:49 +01:00
..
__init__.py
test_modeling_flax_xglm.py
test_modeling_xglm.py [XGLM] run sampling test on CPU to be deterministic (#15892) 2022-03-02 17:55:49 +01:00
test_tokenization_xglm.py