Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3
4model_id = "liamka/sf-100"
5tokenizer = AutoTokenizer.from_pretrained(model_id)
6model = AutoModelForCausalLM.from_pretrained(
7 model_id,
8 torch_dtype=torch.bfloat16,
9 device_map="auto",
10)
11
12messages = [{"role": "user", "content": "Hi, who are you?"}]
13inputs = tokenizer.apply_chat_template(
14 messages, add_generation_prompt=True, return_tensors="pt"
15).to(model.device)
16
17out = model.generate(inputs, max_new_tokens=256, do_sample=True, temperature=0.7, top_p=0.9)
18print(tokenizer.decode(out[0][inputs.shape[-1]:], skip_special_tokens=True))SFTTrainer)