1from transformers import AutoModelForCausalLM, AutoTokenizer
2from peft import PeftModel
3import torch
4
5base_model = AutoModelForCausalLM.from_pretrained(
6 "Qwen/Qwen2.5-3B-Instruct",
7 torch_dtype=torch.float16,
8 device_map="auto",
9)
10model = PeftModel.from_pretrained(base_model, "bhaiyahnsingh45/bhaiya-loan-assistant-lora")
11model.eval()
12
13tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2.5-3B-Instruct")
14
15messages = [
16 {"role": "system", "content": "You are a compliant and helpful loan assistant for Bhaiya & Company — Banking & Finance Division."},
17 {"role": "user", "content": "My salary is 55,000, CIBIL 740, age 30. Am I eligible for a personal loan?"},
18]
19
20inputs = tokenizer.apply_chat_template(messages, tokenize=True, add_generation_prompt=True, return_tensors="pt").to(model.device)
21
22with torch.no_grad():
23 outputs = model.generate(
24 input_ids=inputs["input_ids"],
25 max_new_tokens=512,
26 temperature=0.3,
27 do_sample=True
28 )
29
30generated_tokens = outputs[0][inputs["input_ids"].shape[1]:]
31
32print(tokenizer.decode(generated_tokens, skip_special_tokens=True))
Trained on
bhaiyahnsingh45/bhaiya-loan-assistant-dataset — a custom instruction-tuning dataset in chat format (system/user/assistant messages).