Views
No views yet
Bạn là một chuyên gia toán học. Hãy giải bài toán sau bằng tiếng Việt.
1from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3
4model_name = "PeterPaker123/Qwen2.5-7B-ViMetaMathQA-Mini"
5
6tokenizer = AutoTokenizer.from_pretrained(model_name)
7model = AutoModelForCausalLM.from_pretrained(
8 model_name,
9 torch_dtype=torch.bfloat16,
10 device_map="auto",
11 attn_implementation="flash_attention_2" # Recommended for H100/A100/L4
12)
13
14messages = [
15 {"role": "system", "content": "Bạn là một chuyên gia toán học. Hãy giải bài toán sau bằng tiếng Việt."},
16 {"role": "user", "content": "Tìm x, biết 2x + 5 = 15."}
17]
18
19text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
20inputs = tokenizer(text, return_tensors="pt").to("cuda")
21
22outputs = model.generate(**inputs, max_new_tokens=512)
23print(tokenizer.decode(outputs[0], skip_special_tokens=True))