Downloads · 30 days
9
6% of all-time downloads
JamieAi33/Phi-2-QLora
Phi-2-QLora is a question answering model from JamieAi33. Use it when the input is a question plus a passage. It is set up for peft. The card lists the license as apache-2.0.
license: apache-2.0 language: - en metrics: - rouge basemodel: - microsoft/phi-2 pipelinetag: question-answering ---
Downloads · 30 days
9
6% of all-time downloads
All-time downloads
152
Public
Repo size
127 MB
Likes
0
Public
Click a slice to open those files.
.safetensors83.9 MB · 66%
From the Hugging Face model README
license: apache-2.0 language:
This repo containes the last checkpoint of my fine tuned model. Click this link to go the final model https://huggingface.co/JamieAi33/Phi-2_PEFT
This model card documents a PEFT-fine-tuned version of microsoft/phi-2 for question-answering tasks. The PEFT fine-tuning improved the model's performance, as detailed in the evaluation section.
microsoft/phi-2The base model microsoft/phi-2 was adapted using Parameter-Efficient Fine-Tuning (PEFT) for question-answering tasks. The training process focused on improving performance metrics while keeping computational costs low.
This model can be used out-of-the-box for question-answering tasks.
The model can be fine-tuned further on domain-specific datasets for improved performance.
Avoid using this model for tasks outside question-answering or where fairness, bias, and ethical considerations are critical without further validation.
Users should be aware that:
Here’s an example of loading the model:
from transformers import AutoModel
from peft import PeftModel
base_model = AutoModel.from_pretrained("microsoft/phi-2")
adapter_model = PeftModel.from_pretrained(base_model, "JamieAi33/Phi-2-QLora")
# Model Name: PEFT Fine-Tuned `microsoft/phi-2`
This repository contains a PEFT fine-tuned version of the `microsoft/phi-2` model for question-answering tasks. The fine-tuning process leveraged Parameter-Efficient Fine-Tuning (PEFT) techniques to achieve improved performance.
---
## Metrics
The model's performance was evaluated using the ROUGE metric. Below are the results:
| **Metric** | **Original Model** | **PEFT Model** | **Absolute Improvement** |
|-----------------|--------------------|----------------|---------------------------|
| **ROUGE-1** | 29.76% | 44.51% | +14.75% |
| **ROUGE-2** | 10.76% | 15.68% | +4.92% |
| **ROUGE-L** | 21.69% | 30.95% | +9.25% |
| **ROUGE-Lsum** | 22.75% | 31.49% | +8.74% |
---
## Training Configuration
| Hyperparameter | Value |
|-----------------------|-------------------------|
| **Batch Size** | 1 |
| **Learning Rate** | 2e-4 |
| **Max Steps** | 1000 |
| **Optimizer** | Paged AdamW (8-bit) |
| **Logging Steps** | 25 |
| **Evaluation Steps** | 25 |
| **Gradient Checkpointing** | Enabled |