I’m trying to finetune BLOOM-350m for a simple classification task. It seems right now this is impossible? Here is the error I’m getting:
Blockquote
File “/usr/local/google/home/windmaple/miniconda3/envs/nlp/lib/python3.7/site-packages/transformers/models/auto/auto_factory.py”, line 448, in from_pretrained
f"Unrecognized configuration class {config.class} for this kind of AutoModel: {cls.name}.\n"
ValueError: Unrecognized configuration class <class ‘transformers.models.bloom.configuration_bloom.BloomConfig’> for this kind of AutoModel: TFAutoModelForSequenceClassification.
Model type should be one of AlbertConfig, BertConfig, CamembertConfig, ConvBertConfig, CTRLConfig, DebertaConfig, DebertaV2Config, DistilBertConfig, ElectraConfig, FlaubertConfig, FunnelConfig, GPT2Config, GPTJConfig, LayoutLMConfig, LongformerConfig, MobileBertConfig, MPNetConfig, OpenAIGPTConfig, RemBertConfig, RobertaConfig, RoFormerConfig, TapasConfig, TransfoXLConfig, XLMConfig, XLMRobertaConfig, XLNetConfig.
Given that BLOOM is so new, I’m curious if this is being worked on?