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 refactor 1/5] Per-folder tests reorganization (#15725)

2022-02-23 15:46:28 -05:00

test_modeling_flax_xglm.py

[Test refactor 1/5] Per-folder tests reorganization (#15725)

2022-02-23 15:46:28 -05:00

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

[Test refactor 1/5] Per-folder tests reorganization (#15725)

2022-02-23 15:46:28 -05:00