Views
No views yet
model-raising/spp-vanilla-1.7b-base.<assistant> marker token (vocabulary 49188).[N.M] citations; response-only loss, one epoch.<|im_start|><assistant>. Use the built-in chat template:1from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3
4repo = "model-raising/spp-vanilla-1.7b-instruct"
5tok = AutoTokenizer.from_pretrained(repo)
6model = AutoModelForCausalLM.from_pretrained(repo, torch_dtype=torch.bfloat16, device_map="auto")
7
8msgs = [{"role": "user", "content": "How should I think about honesty?"}]
9ids = tok.apply_chat_template(msgs, add_generation_prompt=True, return_tensors="pt").to(model.device)
10out = model.generate(ids, max_new_tokens=512)
11print(tok.decode(out[0, ids.shape[1]:], skip_special_tokens=False))