Views
No views yet
### User: and ### Assistant: conversationstrust_remote_code=True.1import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4model_id = "Voyager466920/Raptor"
5tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
6model = AutoModelForCausalLM.from_pretrained(
7 model_id,
8 trust_remote_code=True,
9 torch_dtype=torch.bfloat16,
10 device_map="auto",
11)
12
13messages = [{"role": "user", "content": "What is the capital of France?"}]
14inputs = tokenizer.apply_chat_template(
15 messages,
16 add_generation_prompt=True,
17 return_tensors="pt",
18).to(model.device)
19output = model.generate(
20 inputs,
21 max_new_tokens=128,
22 do_sample=True,
23 temperature=0.7,
24 top_p=0.9,
25)
26print(tokenizer.decode(output[0, inputs.shape[1]:], skip_special_tokens=True))