Views
No views yet
thesreedath/slm-125m-base (125M, LLaMA).
Trained on ~8,000 grounded Q&A pairs (RAFT-style: answer from the provided
context) synthesized with Gemini 2.5 Flash and grounding-judged.<|system|> <|user|> <|assistant|> special tokens.1from transformers import AutoModelForCausalLM, AutoTokenizer
2tok = AutoTokenizer.from_pretrained("sumitguha13/slm125MLIVE-sft")
3model = AutoModelForCausalLM.from_pretrained("sumitguha13/slm125MLIVE-sft")
4prompt = ("<|bos|><|system|>\nYou are a helpful assistant. Answer using only the "
5 "provided context.\n<|user|>\nContext:\n<PASSAGE>\n\nQuestion: <Q>\n<|assistant|>\n")
6ids = tok(prompt, return_tensors="pt", add_special_tokens=False).input_ids
7print(tok.decode(model.generate(ids, max_new_tokens=120, repetition_penalty=1.3)[0], skip_special_tokens=True))