Ошибка CUDA: утверждение на стороне устройства сработало, когда роберта моделирует cuda ()

Question

эта ошибка возникает, когда я хочу загрузить модель Роберта в cuda. Код выглядит следующим образом:

import os 
os.environ['CUDA_LAUNCH_BLOCKING']='1'
device = torch.device("cuda")
model_config = transformers.ElectraConfig.from_pretrained(config.ELECTRA_PATH)
model_config.output_hidden_states = True
bert = transformers.ElectraModel.from_pretrained(config.ELECTRA_PATH, config=model_config)
bert = bert.to(device)

Информация об ошибке выглядит следующим образом: