311 Commits

Author	SHA1	Message	Date
Nicolas Patry	ecb4662d17	Attention mask is important in the case of batching... (#16222) ... * Attention mask is important in the case of batching... * Improve the fix. * Making the sentence different enough that they exhibit different predictions.	2022-03-18 10:02:12 +01:00
Nicolas Patry	f4e4ad34cc	Add ForInstanceSegmentation models to image-segmentation pipelines (#15937) ... * Adding ForInstanceSegmentation to pipelines. * Last fix `category_id` renamed to `label_id`. * Can't be none no more. * No `is_thing_map` anymore.	2022-03-09 10:19:05 +01:00
Nicolas Patry	7ade7c1794	Updating the slow tests: (#15893) ... Linked to https://github.com/huggingface/transformers/pull/15826	2022-03-04 12:32:19 +01:00
Nicolas Patry	a6e3b17981	Re-enabling all fast pipeline tests. (#15924)	2022-03-04 09:53:00 +01:00
Nicolas Patry	3822e4a563	Enabling MaskFormer in pipelines (#15917) ... * Enabling MaskFormer in ppipelines No AutoModel though :( * Ooops local file.	2022-03-03 16:31:41 +01:00
Nicolas Patry	b693cbf99c	The tests were not updated after the addition of torch.diag (#15890) ... in the scoring (which is more correct)	2022-03-03 15:33:49 +01:00
Nicolas Patry	6e57a56987	Adding timestamps for CTC with LM in ASR pipeline. (#15863) ... * Adding timestamps for CTC with LM in ASR pipeline. * iRemove print. * Nit change.	2022-03-02 10:49:05 +01:00
Nicolas Patry	97f9b8a27b	Fixing the timestamps with chunking. (#15843) ... * Fixing the timestamps with chunking. * The changes modified (and fixed) the striding tests. * Adding a tokenizer test. * Update src/transformers/pipelines/automatic_speech_recognition.py Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> * Defense -> comment. * Update src/transformers/models/wav2vec2/tokenization_wav2vec2.py Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>	2022-02-28 21:00:21 +01:00
Nicolas Patry	ad0d7d1745	Adding the option to return_timestamps on pure CTC ASR models. (#15792) ... * Adding the option to return_timestamps on pure CTC ASR models. * Remove `math.prod` which was introduced in Python 3.8 * int are not floats. * Reworking the PR to support "char" vs "word" output. * Fixup! * Update src/transformers/pipelines/automatic_speech_recognition.py Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> * Update src/transformers/pipelines/automatic_speech_recognition.py Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> * Update src/transformers/pipelines/automatic_speech_recognition.py Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> * Update src/transformers/pipelines/automatic_speech_recognition.py Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> * Update src/transformers/pipelines/automatic_speech_recognition.py Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> * Update src/transformers/pipelines/automatic_speech_recognition.py Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> * Update src/transformers/pipelines/automatic_speech_recognition.py Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> * Update src/transformers/pipelines/automatic_speech_recognition.py Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> * Update src/transformers/pipelines/automatic_speech_recognition.py Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> * Quality. Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>	2022-02-25 14:06:45 +01:00
Sylvain Gugger	074645e32a	Fix semantic segmentation pipeline test (#15826)	2022-02-25 09:21:29 +01:00
Lysandre Debut	29c10a41d0	[Test refactor 1/5] Per-folder tests reorganization (#15725) ... * Per-folder tests reorganization Co-authored-by: sgugger <sylvain.gugger@gmail.com> Co-authored-by: Stas Bekman <stas@stason.org>	2022-02-23 15:46:28 -05:00