Kernel dies when trying to execute the model

by ivanalexo - opened

Hi, I got this error when trying to download the model, all the process seems to complete, but then the error occurs
when I run this code:

from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("microsoft/BioGPT-Large-PubMedQA")
model = AutoModelForCausalLM.from_pretrained("microsoft/BioGPT-Large-PubMedQA")

In jupyter the error says that the kernel died, when I execute again the same happens, I tried in Google Colab as well and the error says that the RAM was comsumed completely
Am I doing wrong something?
I'll appreciate your answer.

Sign up or log in to comment