text_plain_output_1.png
from transformers import AutoTokenizer, OPTForCausalLM

from transformers import AutoTokenizer, OPTForCausalLM
model = OPTForCausalLM.from_pretrained('facebook/opt-350m')
tokenizer = AutoTokenizer.from_pretrained('facebook/opt-350m')
prompt = 'Hey, are you consciours? Can you talk to me?'

generate(prompt)