valuesimplex / FinBERT

Apache License 2.0
677 stars 110 forks source link

下载的模型的config.json文件中缺少model_type #13

Open pymask opened 2 years ago

pymask commented 2 years ago

代码:model = AutoModelForSequenceClassification.from_pretrained(checkpoint) 运行报错缺少配置文件 ValueError: Unrecognized model in XXX. Should have a model_type key in its config.json, or contain one of the following strings in its name: albert, bart, beit, bert, bert-generation, big_bird, bigbird_pegasus, blenderbot, blenderbot-small, bloom, camembert, canine, clip, codegen, convbert, convnext, ctrl, cvt, data2vec-audio, data2vec-text, data2vec-vision, deberta, deberta-v2, decision_transformer, deit, detr, distilbert, dpr, dpt, electra, encoder-decoder, flaubert, flava, fnet, fsmt, funnel, glpn, gpt2, gpt_neo, gpt_neox, gptj, groupvit, hubert, ibert, imagegpt, layoutlm, layoutlmv2, layoutlmv3, led, levit, longformer, longt5, luke, lxmert, m2m_100, marian, maskformer, mbart, mctct, megatron-bert, mobilebert, mobilevit, mpnet, mt5, mvp, nezha, nystromformer, openai-gpt, opt, owlvit, pegasus, perceiver, plbart, poolformer, prophetnet, qdqbert, rag, realm, reformer, regnet, rembert, resnet, retribert, roberta, roformer, segformer, sew, sew-d, speech-encoder-decoder, speech_to_text, speech_to_text_2, splinter, squeezebert, swin, t5, tapas, trajectory_transformer, transfo-xl, trocr, unispeech, unispeech-sat, van, vilt, vision-encoder-decoder, vision-text-dual-encoder, visual_bert, vit, vit_mae, wav2vec2, wav2vec2-conformer, wavlm, xglm, xlm, xlm-prophetnet, xlm-roberta, xlm-roberta-xl, xlnet, yolos, yoso

Sunlight5288 commented 2 years ago

你把这个模型的文件夹改为以小写英文"bert"开头的名称即可,如"bert-finance-L-12_H-768_A-12_pytorch",同时文件夹里面的json文件改为:config.json