Views
No views yet
category (8 classes), urgency, and entities (order_id, product_name, account_email, sentiment)1import torch
2from peft import PeftModel
3from transformers import AutoModelForCausalLM, AutoTokenizer
4
5model = AutoModelForCausalLM.from_pretrained(
6 "Qwen/Qwen2.5-3B-Instruct",
7 torch_dtype=torch.bfloat16,
8 device_map="auto",
9)
10model = PeftModel.from_pretrained(model, "abuzarkhan/bilingual-ticket-triage-adapter")
11
12tokenizer = AutoTokenizer.from_pretrained("abuzarkhan/bilingual-ticket-triage-adapter")
13
14messages = [{"role": "user", "content": "mera order kab tak aayega? bohot late ho raha hai"}]
15inputs = tokenizer.apply_chat_template(messages, tokenize=True, return_dict=True, return_tensors="pt").to("cuda")
16out = model.generate(**inputs, max_new_tokens=256)
17print(tokenizer.decode(out[0][inputs["input_ids"].shape[1]:], skip_special_tokens=True))notebooks/qwen_qlora_training.ipynb)results/.