Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2import torch
3
4path = "outputs/checkpoint/step-78100-hf2"
5tok = AutoTokenizer.from_pretrained(path)
6model = AutoModelForCausalLM.from_pretrained(path, dtype="auto", device_map="auto")
7
8prompt = "Why is the sky blue?"
9enc = tok(prompt, return_tensors="pt", add_special_tokens=False).to(model.device)
10out = model.generate(**enc, max_new_tokens=64, do_sample=True, temperature=0.8)
11print(tok.decode(out[0], skip_special_tokens=False))add_special_tokens=False when tokenizing prompts for generation.model.safetensors.index.json and re-save.