LiteGPT Project

This repository contains the code and resources for training a Small Language Model (SLM) from scratch, based on Andrej Karpathy's nanoGPT.

The goal of this project is to build a GPT-2 (124M parameter) equivalent model trained on high-quality educational data and subsequently fine-tuned for instruction following.

Project Overview

• Architecture: GPT-2 Small (12 layers, 12 heads, 768 embedding dim)
• Dataset: FineWeb-Edu (Sample 10B tokens)
• Fine-Tuning: Alpaca Dataset (Instruction Tuning)
• Framework: PyTorch (nanoGPT)

Setup

1. Install Dependencies

   pip install -r requirements.txt

2. Prepare Data The training data is processed using the scripts in the data/ directory.

   Note: The actual datasets (FineWeb-Edu and Alpaca) are not included in this repository to keep it light. You can download and prepare them using the provided prepare.py scripts in the respective data/ subdirectories.

Usage

1. Training

To train the base model:

python train.py config/train_fineweb.py

2. Instruction Fine-Tuning

To fine-tune the model on instructions:

python train.py config/finetune_alpaca.py

3. Inference / Chat

We provide interactive scripts to test the models:

• Check Base Model:

  python check_model.py

  Interactively prompts the base model (trained on FineWeb-Edu).

• Check Instruction Model:

  python check_instruct.py

  Interactively chats with the instruction-tuned model. Supports commands like /temp, /tokens.

4. Publish to Hugging Face

To export your trained model to the Hugging Face Hub:

python export_to_hub.py --checkpoint out-instruct/ckpt.pt --repo_name <your-username>/<your-model-name>

Note: Make sure you remain logged in via huggingface-cli login or provide a token.

5. Use from Hugging Face

We host two versions of the model:

• Instruct Model: koganrath/LiteGPT-Instruct (Fine-tuned on Alpaca)
• Base Model: koganrath/LiteGPT-Base (Pre-trained on FineWeb)

We provide a standalone inference setup in the hugging_face_inference/ directory.

To use it:

cd hugging_face_inference
pip install -r requirements.txt

# Run Instruct Model
python inference_instruct.py

# Run Base Model
python inference_base.py

Or in Python:

from transformers import GPT2LMHeadModel, GPT2Tokenizer

model = GPT2LMHeadModel.from_pretrained("koganrath/LiteGPT")
tokenizer = GPT2Tokenizer.from_pretrained("gpt2")

Directory Structure

• data/: Data preparation scripts and raw data.
• config/: Configuration files for training and fine-tuning.
• out-fineweb/: Checkpoints and logs for the base model.
• out-instruct/: Checkpoints for the instruction-tuned model.
• hugging_face_inference/: Standalone scripts for using the hosted model.
• check_model.py: Inference script for the base model.
• check_instruct.py: Inference script for the instruct model.
• export_to_hub.py: Script to export checkpoints to Hugging Face.

Acknowledgements

• nanoGPT by Andrej Karpathy
• FineWeb-Edu Dataset
• Alpaca Dataset

Testing Environment

The code has been tested on Windows with Python 3.11.9.

For a complete list of installed packages in the testing environment (including specific versions), see test_env_requirements.txt.