transformers/tests/models/bloom
Younes Belkada 587d84b178
Add BloomForQuestionAnswering (#19310)
* add bloom for question answering

- attempt to add Bloom for question answering
- adapted from `GPTJForQuestionAnswering`
- Fixed `num_labels` to `2` for common tests
- Added a bit of docstring
- All common tests pass

* Update src/transformers/models/bloom/modeling_bloom.py

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

* revert changes related to `num_labels`

Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
2022-10-04 17:52:13 +02:00
..
__init__.py BLOOM (#17474) 2022-06-09 12:00:40 +02:00
test_modeling_bloom.py Add BloomForQuestionAnswering (#19310) 2022-10-04 17:52:13 +02:00
test_tokenization_bloom.py BLOOM minor changes on tokenizer (#17823) 2022-06-23 15:57:12 +02:00