AutoModelForSequenceClassification требует библиотеки PyTorch, но она не была найдена в вашей среде.

Question

Я пытаюсь использовать трансформатор Роберта и предварительно обученную модель, но продолжаю получать эту ошибку:

          ImportError: 
AutoModelForSequenceClassification requires the PyTorch library but it was not found in your environment. Checkout the instructions on the
installation page: https://pytorch.org/get-started/locally/ and follow the ones that match your environment.

Вот мой код:

      # Tasks:
# emoji, emotion, hate, irony, offensive, sentiment
# stance/abortion, stance/atheism, stance/climate, stance/feminist, stance/hillary

task='sentiment'
MODEL = f"cardiffnlp/twitter-roberta-base-{task}"

tokenizer = AutoTokenizer.from_pretrained(MODEL)
# download label mapping
labels=[]
mapping_link = f"https://raw.githubusercontent.com/cardiffnlp/tweeteval/main/datasets/{task}/mapping.txt"
model = AutoModelForSequenceClassification.from_pretrained(MODEL)
model.save_pretrained(MODEL)
labels=[]
mapping_link = f"https://raw.githubusercontent.com/cardiffnlp/tweeteval/main/datasets/{task}/mapping.txt"
with urllib.request.urlopen(mapping_link) as f:
    html = f.read().decode('utf-8').split("\n")
    csvreader = csv.reader(html, delimiter='\t')
labels = [row[1] for row in csvreader if len(row) > 1]

Убедился, что PyTorch установлен и работает:

Answer 1

У меня была такая же проблема. Решилось для меня перезапуском ядра.

Answer 2

Вам необходимо сначала установить Pytorch, прежде чем импортировать HuggingFace Transformer.

В противном случае вам придется перезапустить ядро