tests/models/bloom/test_modeling_bloom.py · 8b6bba54a707a718149f11c2b0631e6ccd715fba · chenpangpang / transformers

"test/vscode:/vscode.git/clone" did not exist on "3da18f30c7078e07a46472c2494abc9f86751e32"

Add `BloomForQuestionAnswering` (#19310) · 587d84b1

Younes Belkada authored Oct 04, 2022



* 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>

587d84b1

test_modeling_bloom.py 33.3 KB

Replace test_modeling_bloom.py