无法从“变压器”导入“AutoModelForSequenceClassification”

Question

不能从“变压器”导入“AutoModelForSequenceClassification”

代码是

from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline

t = AutoTokenizer.from_pretrained('/some/directory')
m = AutoModelForSequenceClassification.from_pretrained('/some/directory')
c2 = pipeline(task = 'sentiment-analysis', model=m, tokenizer=t)

错误是

cannot import 'AutoModelForSequenceClassification' from 'transformers'

Answer 1

​

from transformers import AutoModelForSequenceClassification, BertForSequenceClassification
from transformers import (XLMRobertaConfig, XLMRobertaTokenizer, TFXLMRobertaModel)            
from transformers import AutoTokenizer, AutoConfig, TFAutoModel    

PRETRAINED_MODEL_TYPES = {
    'xlmroberta': (AutoConfig, AutoModelForSequenceClassification, AutoTokenizer, 'akhooli/xlm-r-large-arabic-toxic')
}

# model_class,model = AutoModelForSequenceClassification.from_pretrained("akhooli/xlm-r-large-arabic-toxic")

config_class, model_class, tokenizer_class, model_name = PRETRAINED_MODEL_TYPES['xlmroberta']

# Download vocabulary from huggingface.co and cache.
tokenizer = AutoTokenizer.from_pretrained(model_name,use_fast=False) #fast tokenizer

tokenizer

​