Views
No views yet
Detoxify was used to filter harmful generations during evaluation.
1from transformers import AutoTokenizer, AutoModelForCausalLM
2from peft import PeftModel
3
4# Load base model and LoRA adapter
5tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen1.5-1.8B-Chat")
6base_model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen1.5-1.8B-Chat")
7model = PeftModel.from_pretrained(base_model, "522H0134-NguyenNhatHuy/Qwen-1.5-1.8B-Chat-SFT")
8
9# Chat example
10prompt = "Xin chào, bạn có thể giúp tôi học tiếng Anh không?"
11inputs = tokenizer(prompt, return_tensors="pt")
12outputs = model.generate(**inputs, max_new_tokens=150)
13print(tokenizer.decode(outputs[0], skip_special_tokens=True))