tests/pipelines/test_pipelines_text_generation.py · fa7f3cf336eb5d93cfaa7611723c299e7851fb02 · chenpangpang / transformers

"tests/nystromformer/test_modeling_nystromformer.py" did not exist on "8bf7312654d40cea1a399e86f9fe8e39e1ea3a1e"

[tests] enable test_pipeline_accelerate_top_p on XPU (#29309) · fa7f3cf3

Fanli Lin authored Mar 05, 2024



* use torch_device

* Update tests/pipelines/test_pipelines_text_generation.py
Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>

* fix style

---------
Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>

fa7f3cf3

test_pipelines_text_generation.py 18 KB

Replace test_pipelines_text_generation.py