Kernel dies when trying to execute the model
#5
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.
Regards