Views
No views yet

1import transformers
2import torch
3
4if __name__ == "__main__":
5 model_name = "abacaj/phi-2-super"
6 tokenizer = transformers.AutoTokenizer.from_pretrained(model_name)
7
8 model = (
9 transformers.AutoModelForCausalLM.from_pretrained(
10 model_name,
11 )
12 .to("cuda:0")
13 .eval()
14 )
15
16 messages = [
17 {"role": "user", "content": "Hello, who are you?"}
18 ]
19 inputs = tokenizer.apply_chat_template(messages, return_tensors="pt").to(model.device)
20 input_ids_cutoff = inputs.size(dim=1)
21
22 with torch.no_grad():
23 generated_ids = model.generate(
24 input_ids=inputs,
25 use_cache=True,
26 max_new_tokens=512,
27 temperature=0.2,
28 top_p=0.95,
29 do_sample=True,
30 eos_token_id=tokenizer.eos_token_id,
31 pad_token_id=tokenizer.pad_token_id,
32 )
33
34 completion = tokenizer.decode(
35 generated_ids[0][input_ids_cutoff:],
36 skip_special_tokens=True,
37 )
38
39 print(completion)1text = "<|endoftext|>[INST] What is your favourite condiment? [/INST]"
2"Well, I'm quite partial to a good squeeze of fresh lemon juice. It adds just the right amount of zesty flavour to whatever I'm cooking up in the kitchen!<|endoftext|> "
3"[INST] Do you have mayonnaise recipes? [/INST]"1 messages = [
2 {"role": "user", "content": "Hello, who are you?"},
3 {"role": "assistant": "content": "I am ..."}
4 ]
5 inputs = tokenizer.apply_chat_template(messages, return_tensors="pt").to(model.device)
