无法从"转换器"导入"自动建模序列分类"

Question

无法从"transformers"导入"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