Как исправить ошибку AttributeError: объект «список» не имеет атрибута «размер»? ⇐ Python

[Anonymous]

Код: Выделить всё

from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
from datasets import load_dataset
from torch.utils.data import DataLoader
import torch
from tqdm import tqdm

# Загрузка модели и токенизатора
model_name = "google/flan-t5-base"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSeq2SeqLM.from_pretrained(model_name)

# Загрузка и подготовка данных
def load_and_prepare_data(dataset_name="squad"):
dataset = load_dataset(dataset_name, split='train')
dataset = dataset.filter(lambda example: len(example['context'])  66 train_model(model, train_dataloader, test_dataloader)
67
68 # Сохранение модели

/usr/local/lib/python3.10/dist-packages/transformers/models/t5/modeling_t5.py in forward(self, input_ids, attention_mask, encoder_hidden_states, encoder_attention_mask, inputs_embeds, head_mask, cross_attn_head_mask, past_key_values, use_cache, output_attentions, output_hidden_states, return_dict)
999             )
1000         elif input_ids is not None:
->  1001             input_shape = input_ids.size()
1002             input_ids = input_ids.view(-1, input_shape[-1])
1003         elif inputs_embeds is not None:

AttributeError: 'list' object has no attribute 'size'

Я так и не понял, в чем ошибка.

Подробнее здесь: https://stackoverflow.com/questions/791 ... ibute-size