tests/xglm/test_modeling_xglm.py · 130b987880a9b1ade5c76dc1413c12c8924fda50 · chenpangpang / transformers · GitLab

Find file Blame History Permalink

[XGLM] run sampling test on CPU to be deterministic (#15892) · 130b9878
Suraj Patil authored Mar 02, 2022
```
* run sampling test on CPU to be deterministic

* input_ids on CPU
```
130b9878

test_modeling_xglm.py 19.1 KB