Views
No views yet
article → input texthighlights → target summary| Model | ROUGE-1 | ROUGE-2 | ROUGE-L |
|---|---|---|---|
| GPT-2 (baseline) | ~0.16 | ~0.09 | ~0.12 |
| Phi-2 (baseline) | ~0.17 | ~0.096 | ~0.13 |
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model = AutoModelForCausalLM.from_pretrained("prasanna030/gpt2-qlora-summarizer")
4tokenizer = AutoTokenizer.from_pretrained("prasanna030/gpt2-qlora-summarizer")
5
6💡 Intended Use
7News article summarization
8Content condensation
9Educational demos for fine-tuning LLMs
10Low-resource NLP experimentation
11
12⚠️ Limitations
13GPT-2 is not inherently optimized for summarization
14May generate:
15repetitive text
16incomplete summaries
17Performance limited due to:
18small dataset subset
19lightweight training
20
21🚫 Out-of-Scope Use
22Not suitable for:
23critical decision-making
24medical/legal summarization
25factual verification tasks
26
27🧠 Key Insight
28
29This project demonstrates that:
30
31Parameter-efficient fine-tuning methods like QLoRA can significantly improve model performance even on limited hardware.
32
33📦 Training Environment
34Platform: Google Colab
35GPU: T4
36Libraries:
37transformers
38peft
39datasets
40bitsandbytes
41
42📜 License
43
44This model follows the license of the base model (GPT-2).
45
46🙌 Acknowledgements
47Hugging Face 🤗
48CNN/DailyMail Dataset
49Open-source LLM community
50
51📬 Contact
52
53Developed by Prasanna Nagarale