riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-29 01:02:25 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity
a81fe4e1df
transformers/tests/pipelines
History
Younes Belkada f83942684d
[pipeline] A simple fix for half-precision & 8bit models (#21479) 
* v1 fix

* adapt from suggestions

* make style

* fix tests

* add gpu tests

* update docs

* fix other tests

* Apply suggestions from code review

Co-authored-by: Nicolas Patry <patry.nicolas@protonmail.com>

* better fix

* make fixup

* better example

* revert changes

* proposal

* more elegant solution

* Update src/transformers/pipelines/automatic_speech_recognition.py

Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>

---------

Co-authored-by: Nicolas Patry <patry.nicolas@protonmail.com>
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
2023-02-10 10:26:17 +01:00
..
__init__.py [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
test_pipelines_audio_classification.py Pipeline testing - using tiny models on Hub (#20426) 2023-01-30 10:39:43 +01:00
test_pipelines_automatic_speech_recognition.py Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
test_pipelines_common.py Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
test_pipelines_conversational.py Pipeline testing - using tiny models on Hub (#20426) 2023-01-30 10:39:43 +01:00
test_pipelines_depth_estimation.py Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
test_pipelines_document_question_answering.py Fix some pipeline tests (#21401) 2023-02-02 19:03:31 +01:00
test_pipelines_feature_extraction.py Pipeline testing - using tiny models on Hub (#20426) 2023-01-30 10:39:43 +01:00
test_pipelines_fill_mask.py Pipeline testing - using tiny models on Hub (#20426) 2023-01-30 10:39:43 +01:00
test_pipelines_image_classification.py Fix some pipeline tests (#21401) 2023-02-02 19:03:31 +01:00
test_pipelines_image_segmentation.py Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
test_pipelines_image_to_text.py Fix some pipeline tests (#21401) 2023-02-02 19:03:31 +01:00
test_pipelines_object_detection.py Fix some pipeline tests (#21401) 2023-02-02 19:03:31 +01:00
test_pipelines_question_answering.py Pipeline testing - using tiny models on Hub (#20426) 2023-01-30 10:39:43 +01:00
test_pipelines_summarization.py Pipeline testing - using tiny models on Hub (#20426) 2023-01-30 10:39:43 +01:00
test_pipelines_table_question_answering.py Add require_torch to 2 pipeline tests (#20585) 2022-12-05 16:06:39 +01:00
test_pipelines_text_classification.py Pipeline testing - using tiny models on Hub (#20426) 2023-01-30 10:39:43 +01:00
test_pipelines_text_generation.py [pipeline] A simple fix for half-precision & 8bit models (#21479) 2023-02-10 10:26:17 +01:00
test_pipelines_text2text_generation.py Pipeline testing - using tiny models on Hub (#20426) 2023-01-30 10:39:43 +01:00
test_pipelines_token_classification.py Pipeline testing - using tiny models on Hub (#20426) 2023-01-30 10:39:43 +01:00
test_pipelines_translation.py Pipeline testing - using tiny models on Hub (#20426) 2023-01-30 10:39:43 +01:00
test_pipelines_video_classification.py Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
test_pipelines_visual_question_answering.py Pipeline testing - using tiny models on Hub (#20426) 2023-01-30 10:39:43 +01:00
test_pipelines_zero_shot_image_classification.py Pipeline testing - using tiny models on Hub (#20426) 2023-01-30 10:39:43 +01:00
test_pipelines_zero_shot_object_detection.py Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
test_pipelines_zero_shot.py Pipeline testing - using tiny models on Hub (#20426) 2023-01-30 10:39:43 +01:00