Usage
from transformers import AutoTokenizer, AutoModelForCausalLM
tok = AutoTokenizer.from_pretrained('facebook/opt-350m')
model = AutoModelForCausalLM.from_pretrained('prasanna2003/opt-350m-instruct')
system_message = "You are AI language model helps the human."
input_prompt = "Define data science."
prompt = '<system>' + system_message + '<human>' + input_prompt + '<assistant>'
prompt = tokenizer(prompt, return_tensors='pt')
out = model.generate(**prompt, max_length=120)
print(tok.decode(out[0]))
- Downloads last month
- 86
Inference Providers
NEW
This model is not currently available via any of the supported third-party Inference Providers, and
the model is not deployed on the HF Inference API.