Mistral-7B QLoRA Fine-Tuned (Retail Banking Chatbot)

Overview

This model is a fine-tuned version of Mistral-7B using QLoRA for domain-specific conversational tasks in retail banking.

Model Details

  • Base Model: mistralai/Mistral-7B
  • Fine-tuning Method: QLoRA (4-bit NF4 quantization)
  • Framework: Hugging Face Transformers + PEFT
  • Task: Banking chatbot / conversational QA

Training

  • Dataset: Custom retail banking dataset
  • Approach: Parameter-Efficient Fine-Tuning (PEFT)
  • Objective: Improve domain-specific response generation

Evaluation

  • ROUGE-1: 0.47
  • ROUGE-L: 0.34
  • BLEU: 0.11

Usage

from transformers import AutoTokenizer, AutoModelForCausalLM
from peft import PeftModel

model_id = "iamnotpalak/mistral-qlora-banking"

tokenizer = AutoTokenizer.from_pretrained(model_id)
base_model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B")

model = PeftModel.from_pretrained(base_model, model_id)
Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for iamnotpalak/mistral-qlora-banking

Finetuned
(489)
this model

Dataset used to train iamnotpalak/mistral-qlora-banking