Hailey Schoelkopf edb672ac5e Add BloomForSequenceClassification and BloomForTokenClassification classes (#17639) ... * add new bloom classes * (feat) add bloom classification tests; make style * style: change import in test * add some typehints to bloom classes * merge main into branch * fix: input checking in bloom seq classification * fix tests * change model class tests * fix few tests - more tests should pass - one test left * make token classifier return hidden states * style: make BLOOM typehints consistent Co-authored-by: Younes Belkada <49240599+younesbelkada@users.noreply.github.com> Co-authored-by: younesbelkada <younesbelkada@gmail.com> Co-authored-by: Younes Belkada <49240599+younesbelkada@users.noreply.github.com>		2022-06-14 17:10:12 +02:00
..
__init__.py	BLOOM (#17474)	2022-06-09 12:00:40 +02:00
test_modeling_bloom.py	Add BloomForSequenceClassification and BloomForTokenClassification classes (#17639)	2022-06-14 17:10:12 +02:00
test_tokenization_bloom.py	BLOOM (#17474)	2022-06-09 12:00:40 +02:00