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

"pytorch_transformers/tests/fixtures/input.txt" did not exist on "f8276008dfaa736a89eae2c9a1bd515841159383"

Fix FP16 inference in TextGenerationPipeline (#20913) · fe65657d

bofeng huang authored Dec 29, 2022



* add torch_dtype attribute to Pipeline

* Use torch_dtype to cast input tensor type in AutomaticSpeechRecognitionPipeline

* Fix code quality

* Add TextGenerationPipeline fp16 test

* Fix code quality

* Remove useless require in tests
Co-authored-by: Nicolas Patry <patry.nicolas@protonmail.com>
Co-authored-by: Nicolas Patry <patry.nicolas@protonmail.com>

fe65657d

test_pipelines_text_generation.py 12.2 KB

Replace test_pipelines_text_generation.py