Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2tok = AutoTokenizer.from_pretrained("thesreedath/slm-125m-sft")
3model = AutoModelForCausalLM.from_pretrained("thesreedath/slm-125m-sft")
4prompt = ("<|bos|><|system|>\nYou are a legal and financial assistant. Answer the "
5 "question using ONLY the provided context.<|eos|>\n<|user|>\n<context>\n"
6 "{PASSAGE}\n</context>\n\nQuestion: {QUESTION}<|eos|>\n<|assistant|>\n")