Views
No views yet

<|im_start|>system
Разговор помеѓу љубопитен корисник и асистент со вештачка интелигенција. Асистентот дава корисни, детални и љубезни одговори на прашањата на корисникот.<|im_end|>
<|im_start|>user
Која планета е позната како 'Црвената Планета'?<|im_end|>
<|im_start|>assistant
Марс<|im_end|> tokenizer.apply_chat_template() method:1messages = [
2 {"role": "system", "content": "Разговор помеѓу љубопитен корисник и асистент со вештачка интелигенција. Асистентот дава корисни, детални и љубезни одговори на прашањата на корисникот."},
3 {"role": "user", "content": "Која планета е позната како 'Црвената Планета'?"}
4]
5gen_input = tokenizer.apply_chat_template(messages,
6 tokenize=True,
7 return_dict=True,
8 return_tensors="pt",
9 add_generation_prompt=True).to("cuda")
10with torch.no_grad():
11 generated_ids = model.generate(**gen_input, max_new_tokens=150,
12 do_sample=True,
13 temperature=0.1,
14 repetition_penalty=1.1,
15 )
16print(tokenizer.decode(generated_ids[0][prompt["input_ids"].shape[1]:], skip_special_tokens=False))