transformers
Add `BloomForSequenceClassification` and `BloomForTokenClassification` classes
#17639
Merged
Merged
Add `BloomForSequenceClassification` and `BloomForTokenClassification` classes #17639
younesbelkada
merged
14 commits into
huggingface:main
from
haileyschoelkopf:bloom_classification_heads
add new bloom classes
35996fc7
(feat) add bloom classification tests; make style
56b5c167
style: change import in test
ccba202a
add some typehints to bloom classes
bf546eea
sgugger
approved these changes
on 2022-06-13
merge main into branch
5656360f
fix merge conflicts
8c0ab2cc
fix: input checking in bloom seq classification
4134e49e
fix tests
8402089f
change model class tests
bfc6d49e
Merge branch 'main' into bloom_classification_heads
331ad3c0
fix few tests
d9fc6fbf
make token classifier return hidden states
a41b278e
Merge branch 'bloom_classification_heads' of https://github.com/haile…
1d591403
younesbelkada
commented
on 2022-06-14
younesbelkada
commented
on 2022-06-14
younesbelkada
approved these changes
on 2022-06-14
style: make BLOOM typehints consistent
194996e2
younesbelkada
changed the title 
Login to write a write a comment.
Login via GitHub