Model Card for Model ID

LoRA-Adapted GPT2-Distilled Model

Model Description

This model is a LoRA-adapted version of distilled GPT2, fine-tuned on the Alpaca dataset to enhance its instruction-following capabilities. The model uses Low-Rank Adaptation (LoRA) to efficiently fine-tune the base model while maintaining its core capabilities and adding new instruction-following abilities.

Model Details

  • Developed by: Shahid Mohiuddin
  • Model type: LoRA-adapted GPT2 (Instruction-tuned Language Model)
  • Language(s): English
  • License: Apache 2.0
  • Finetuned from model: distilbert/distilgpt2

Uses

Direct Use

This model is designed for instruction-following tasks and can be used for:

  • Creative writing and storytelling
  • Explanatory content generation
  • Question answering
  • Task-based instructions

The model shows particular improvements in:

  • Narrative coherence
  • Contextual understanding
  • Structured reasoning

Code Example

from transformers import AutoModelForCausalLM, AutoTokenizer
from peft import PeftModel

base_model = AutoModelForCausalLM.from_pretrained("distilbert/distilgpt2")
model = PeftModel.from_pretrained(base_model, "shahidmo/gpt2-distilled-lora-alpaca")

Training Details

Training Data

  • Dataset: Alpaca 52k instructions dataset
  • Preprocessing: Standard text preprocessing with instruction-response format

Training Procedure

  • Training Type: LoRA (Low-Rank Adaptation)
  • Hardware: A4000 16GB GPU

Model Architecture

  • Base Model: distilbert/distilgpt2
  • Adaptation Method: LoRA
  • Parameter-Efficient Fine-Tuning: Used LoRA to minimize training parameters while maximizing adaptation effectiveness

Evaluation

Example Outputs

The model shows significant improvements over the base model in several areas:

  1. Creative Writing Example: Prompt: "Write a short story about a magical key."

    Base Model: The Key of My Life is the Magic Ring!

    LoRA-tuned Model: A magic secret was discovered in the garden of an ancient Egyptian temple, hidden beneath its walls and deep into their own soil. The mystery behind the mysterious gemstone is that it's not only due to being buried under water but also because there are otherworldly elements lurking underneath this precious treasure—something which has been revealed by archaeologists at Uppsala University.

  2. Scientific Explanation Example: Prompt: "Explain why leaves change color in autumn using simple terms."

    Base Model: Autumn means a cold winter, with temperatures rising and falling as the seasons cool down on top of each other. Leaves can be used to warm up

    LoRA-tuned Model: Leaves change colors as they grow and mature, creating a more uniform appearance for the season's seasonal seasons. As winter progresses, blueberries will begin

Limitations

While the model shows improvements in instruction following and coherence, users should be aware of these limitations:

  • Limited context window inherited from base GPT2
  • May occasionally generate incomplete or inconsistent responses
  • Scientific explanations may need fact-checking

Citation

If you use this model in your research, please cite:

@misc{gpt2-distilled-lora-alpaca,
  author = {Mohammed Khaja, Shahid Mohiuddin},
  title = {LoRA-Adapted GPT2-Distilled Model},
  year = {2024},
  publisher = {Hugging Face},
  journal = {Hugging Face Model Hub},
  howpublished = {\url{https://huggingface.co./shahidmo/gpt2-distilled-lora-alpaca}}
}

Model Card Contact

For questions about this model, please contact Shahid Mohammed via Hugging Face.

Version 2 of 2

Downloads last month
19
Inference API
Unable to determine this model’s pipeline type. Check the docs .

Model tree for shahidmo99/gpt2-distilled-lora-alpaca

Adapter
(18)
this model

Dataset used to train shahidmo99/gpt2-distilled-lora-alpaca