boringdude/Salvia

AI Programming Assistant

This project trains an AI model to assist with programming-related tasks using PyTorch and Transformers.

Setup Instructions

1. Download the dataset: Ensure you have the dataset zip file from GitHub or Kaggle.
2. Extract the dataset: Use the provided script to extract the dataset.
3. Train the model: Run the training script to train your GPT-2 model on the programming dataset.

How to Use

• Place your dataset zip file in the datasets folder.
• Run main.py to start the download, extraction, and training process.
• Trained models will be saved in the trained_model directory.

Requirements

• Python 3.8+
• Required Python packages (listed in requirements.txt)