Views
No views yet
Qwen/Qwen2.5-3B-Instruct using 4-bit QLoRA.productissuedisputed_amountsummaryproduct and issue are the categories present in the CFPB complaint data.disputed_amount and summary were extracted using deterministic
regex/rule-based methods.| Metric | Baseline | Fine-tuned |
|---|---|---|
| JSON validity | 100.0% | 100.0% |
| Field accuracy | 60.0% | 69.2% |
1from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
2from peft import PeftModel
3import torch
4
5bnb_config = BitsAndBytesConfig(
6 load_in_4bit=True,
7 bnb_4bit_quant_type="nf4",
8 bnb_4bit_compute_dtype=torch.bfloat16
9)
10
11base = AutoModelForCausalLM.from_pretrained(
12 "Qwen/Qwen2.5-3B-Instruct",
13 quantization_config=bnb_config,
14 device_map="auto"
15)
16
17model = PeftModel.from_pretrained(
18 base,
19 "SatyamGhosh/qlora-consumer-complaint-extractor"
20)
21
22tokenizer = AutoTokenizer.from_pretrained(
23 "SatyamGhosh/qlora-consumer-complaint-extractor"
24)