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

[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