Views
No views yet
qwen2.5-3b-vivu-travel-vn is a 3B-parameter Small Language Model (SLM) fine-tuned for the Vietnamese Tourism Domain. Built on Qwen2.5-3B-Instruct using Unsloth (PEFT/LoRA), it acts as ViVu, an intelligent travel assistant optimized for Advanced RAG pipelines.1pip install transformers vllm accelerate
2
3from transformers import AutoModelForCausalLM, AutoTokenizer
4
5model_name = "thanhdo881/qwen2.5-3b-vivu-travel-vn"
6
7model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype="auto", device_map="auto")
8tokenizer = AutoTokenizer.from_pretrained(model_name)
9
10# 1. Prepare RAG Context & Query
11context = "Đà Lạt nằm trên cao nguyên Lâm Viên, nổi tiếng với khí hậu ôn đới và Hồ Xuân Hương."
12question = "Đà Lạt có những đặc điểm gì nổi bật?"
13prompt = f"Dựa vào thông tin sau:\n{context}\n\nHãy trả lời câu hỏi: {question}"
14
15# 2. Build Messages
16messages = [
17 {"role": "system", "content": "Bạn là ViVu, trợ lý du lịch Việt Nam. Chỉ trả lời dựa trên ngữ cảnh được cung cấp."},
18 {"role": "user", "content": prompt}
19]
20
21# 3. Generate
22text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
23inputs = tokenizer([text], return_tensors="pt").to(model.device)
24
25outputs = model.generate(**inputs, max_new_tokens=512, temperature=0.3, repetition_penalty=1.1)
26response = tokenizer.batch_decode(outputs[:, inputs.input_ids.shape[1]:], skip_special_tokens=True)[0]
27
28print(response)