Files
HuggingFace_transformer/docs/source/en
Billy Cao ac26260436 Allow FP16 or other precision inference for Pipelines (#31342)
* cast image features to model.dtype where needed to support FP16 or other precision in pipelines

* Update src/transformers/pipelines/image_feature_extraction.py

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

* Use .to instead

* Add FP16 pipeline support for zeroshot audio classification

* Remove unused torch imports

* Add docs on FP16 pipeline

* Remove unused import

* Add FP16 tests to pipeline mixin

* Add fp16 placeholder for mask_generation pipeline test

* Add FP16 tests for all pipelines

* Fix formatting

* Remove torch_dtype arg from is_pipeline_test_to_skip*

* Fix format

* trigger ci

---------

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
2024-07-05 17:21:50 +01:00
..
2024-06-23 20:27:21 +01:00
2024-06-27 17:51:42 +02:00
2023-09-04 11:15:12 +01:00
2022-04-04 10:25:46 -04:00
2023-12-20 10:37:23 -08:00
2023-11-13 14:20:54 +01:00
2024-06-27 17:36:19 +02:00
2024-04-18 12:49:43 -04:00
2022-04-04 10:25:46 -04:00
2024-06-10 09:53:25 +01:00
2024-06-12 11:33:00 +01:00
2024-06-26 21:59:08 +01:00