Views
No views yet
1import torch
2import transformers
3
4model = transformers.AutoModelForCausalLM.from_pretrained(
5 "miguelcarv/phi-1_5-slimorca",
6 trust_remote_code=True
7)
8tokenizer = transformers.AutoTokenizer.from_pretrained("microsoft/phi-1_5")
9
10
11SYSTEM_PROMPT = "You are an AI assistant. You will be given a task. You must generate a detailed and long answer."
12input_text = f"""{SYSTEM_PROMPT}
13
14Instruction: Give me the first 5 prime numbers and explain what prime numbers are.
15Output:"""
16
17with torch.no_grad():
18 outputs = model.generate(
19 tokenizer(input_text, return_tensors="pt")['input_ids'],
20 max_length=1024,
21 num_beams = 3,
22 eos_token_id = tokenizer.eos_token_id
23 )
24print(tokenizer.decode(outputs[0], skip_special_tokens=True))