01。
概述
02。
03。
特性
04。
Qwen2.5规格与性能
05。
Qwen2.5的提升
fromtransformersimportAutoModelForCausalLM,AutoTokenizer
model_name="Qwen/Qwen2.5-7B-Instruct"
model=AutoModelForCausalLM.from_pretrained(
model_name,
torch_dtype="auto",
device_map="auto"
)
tokenizer=AutoTokenizer.from_pretrained(model_name)
prompt="Givemeashortintroductiontolargelanguagemodel."
messages=[
{"role":"system","content":"YouareQwen,createdbyAlibabaCloud.Youareahelpfulassistant."},
{"role":"user","content":prompt}
]
text=tokenizer.apply_chat_template(
messages,
tokenize=False,
add_generation_prompt=True
)
model_inputs=tokenizer([text],return_tensors="pt").to(model.device)
generated_ids=model.generate(
**model_inputs,
max_new_tokens=512
)
generated_ids=[
output_ids[len(input_ids):]forinput_ids,output_idsinzip(model_inputs.input_ids,generated_ids)
]
response=tokenizer.batch_decode(generated_ids,skip_special_tokens=True)[0]
06。
结语
| 欢迎光临 链载Ai (https://www.lianzai.com/) | Powered by Discuz! X3.5 |