1.transformers代码
from transformers import AutoTokenizer
path = "./models/Qwen2.5-0.5B-Instruct"
tokenizer = AutoTokenizer.from_pretrained(path)
messages = [{"role": "user", "content": "你好,介绍一下你自己"}]
text = tokenizer.apply_chat_template(
messages, tokenize=False, add_generation_prompt=True
)
print(text)
inputs = tokenizer(text, return_tensors="pt")
print("inputs.input_ids:\n", inputs.input_ids)
print("inputs.input_ids.shape", inputs.input_ids.shape)
print("inputs.attention_mask:\n", inputs.attention_mask)
print("inputs.attention_mask.shape", inputs.attention_mask.shape)

