Killed when trying to execute the code
#1
by
NamburiSrinath
- opened
Hi,
I just pasted this code in a .py file and when I executed the python file, it downloaded the model and threw the output "Killed"
from transformers import pipeline
generator = pipeline('text-generation', model='EleutherAI/gpt-neo-2.7B')
generator("Describe a dog?", do_sample=True, min_length=50)
Attaching a screenshot for reference
You likely ran out of memory.
how much memory does it need to run an inference ?
I think this one was about 20GB or so